TopKScoreConsumer

class TopKScoreConsumer(k: int = 3, device: str | device | None = None)[source]

基础类：ScoreConsumer

收集前k个三元组和分数。

初始化消费者。

Parameters:

  • k (int) – 要收集的得分最高的三元组的数量

  • device (str | device | None) – 模型的设备

属性摘要

flatten

方法总结

__call__(batch, target, scores)

消耗给定 hr_batch 的分数。

属性文档

flatten: bool = False

方法文档

__call__(batch: Tensor, target: Literal['head', 'relation', 'tail'], scores: Tensor) → None[源代码]

消耗给定 hr_batch 的分数。

Parameters:
Return type:

无