Skip to content

Latest commit

 

History

History
7 lines (6 loc) · 323 Bytes

README.md

File metadata and controls

7 lines (6 loc) · 323 Bytes

BERT-based Ranker (w/curriculum learning)

  • 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