AllPredictionDataset

class AllPredictionDataset(num_entities, num_relations, **kwargs)[source]

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__()