PredictionDataset

class PredictionDataset(target: Literal['head', 'relation', 'tail'] = '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.