PredictionDataset¶
- class PredictionDataset(target='tail')[source]¶
Bases:
Dataset
A base class for prediction datasets.
Initialize the dataset.
- Parameters:
target (
Literal
[‘head’, ‘relation’, ‘tail’]) – the prediction target to use. Prefer targets which are efficient to predict with the given model, e.g., tails for ConvE.