calc_ranks

unike.config.calc_ranks(idx: torch.Tensor, label: torch.Tensor, pred_score: torch.Tensor) → torch.Tensor

计算三元组的排名。

参数:

• idx (torch.Tensor) – 需要链接预测的实体 ID

• label (torch.Tensor) – 标签

• pred_score (torch.Tensor) – 三元组的评分

返回:

正确三元组的排名

返回类型:

torch.Tensor