所有预测数据集

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

基础类：PredictionDataset

用于预测所有可能三元组的数据集。

初始化数据集。

Parameters:

• num_entities (int) – 实体的数量

• num_relations (int) – 关系的数量

• kwargs – 传递给 PredictionDataset.__init__() 的额外基于关键字的参数