transformer_rankers.scripts.train_sentenceBERT_crr.CRRDataReader¶
-
class
transformer_rankers.scripts.train_sentenceBERT_crr.
CRRDataReader
(dataset_folder)[source]¶ Bases:
object
Reads in the CRR dataset.
Methods
__init__
(dataset_folder)Initialize self.
get_examples
(filename[, max_examples])filename specified which data split to use (train.csv, dev.csv, test.csv).