head_predict

pybind11_ke.config.head_predict(batch: dict[str, Union[dgl.DGLGraph, torch.Tensor]], model: RGCN) → torch.Tensor

进行头实体的链接预测。

参数:

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

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

返回:

正确三元组的排名

返回类型:

torch.Tensor