预测数据集

class PredictionDataset(target: Literal['head', 'relation', 'tail'] = 'tail')[源代码]

基础类：Dataset

预测数据集的基类。

初始化数据集。

Parameters:: 目标 (Literal['head', 'relation', 'tail']) – 使用的预测目标。优先选择使用给定模型预测效率高的目标，例如，ConvE的尾部。