capreolus.sampler
¶
Package Contents¶
Classes¶
TrainDataset (qid_docid_to_rank, qrels, extractor, relevance_level=1) |
Samples training data. Intended to be used with a pytorch DataLoader |
PredDataset (qid_docid_to_rank, extractor) |
Creates a Dataset for evaluation (test) data to be used with a pytorch DataLoader |
-
class
capreolus.sampler.
TrainDataset
(qid_docid_to_rank, qrels, extractor, relevance_level=1)[source]¶ Bases:
torch.utils.data.IterableDataset
Samples training data. Intended to be used with a pytorch DataLoader