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.