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.CRRDataReaderReader 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).