- selection_model.py: Selection Model
- train_bert_selection.py: Training the bert ranker model
- eval_selection_model.py: Evaluating the bert ranker model and save the prediction result
- preprocess_dataset.py: Download and Preprocess the dataset
- utils.py: Utility functions
-
Notifications
You must be signed in to change notification settings - Fork 1
leenw23/bert_ranker
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
BERT based ranker for dialogue response selection tasks
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published