link_predict

pybind11_ke.config.link_predict(batch: dict[str, Union[dgl.DGLGraph, torch.Tensor]], model: RGCN, prediction: str = 'all') → torch.Tensor

进行链接预测。

参数:

• batch (dict[str, Union[dgl.DGLGraph , torch.Tensor]]) – R-GCN [SKB+18] 的测试数据

• model (RGCN) – R-GCN [SKB+18] 模型

• prediction (str) – “all”, “head”, “tail”

返回:

正确三元组的排名

返回类型:

torch.Tensor