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.

__init__(dataset_folder)[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(dataset_folder)

Initialize self.

get_examples(filename[, max_examples])

filename specified which data split to use (train.csv, dev.csv, test.csv).

get_examples(filename, max_examples=0)[source]

filename specified which data split to use (train.csv, dev.csv, test.csv).