transformer_rankers.scripts.train_sentenceBERT_crr.CRRBenchmarkDataReader¶
-
class
transformer_rankers.scripts.train_sentenceBERT_crr.
CRRBenchmarkDataReader
(dataset_folder)[source]¶ Bases:
transformer_rankers.scripts.train_sentenceBERT_crr.CRRDataReader
Reader especially for CRR (conversation response ranking) datasets.
Methods
__init__
(dataset_folder)Initialize self.
get_examples
(filename[, max_examples])filename specified which data split to use (train.csv, dev.csv, test.csv).