AllPredictionDataset

class AllPredictionDataset(num_entities: int, num_relations: int, **kwargs)

Bases: PredictionDataset

A dataset for predicting all possible triples.

Initialize the dataset.

Parameters:

• num_entities (int) – the number of entities

• num_relations (int) – the number of relations

• kwargs – additional keyword-based parameters passed to PredictionDataset.__init__()