Skip to content

leenw23/bert_ranker_curriculum_learning

Repository files navigation

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

About

BERT-based Ranker with curriculum learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages