分数消费者

class ScoreConsumer[source]

基础类:object

访问者模式得分的消费者。

方法总结

__call__(batch, target, scores)

消耗给定 hr_batch 的分数。

finalize()

最终确定结果以构建一个评分包。

方法文档

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

消耗给定 hr_batch 的分数。

Parameters:
Return type:

finalize() ScorePack[source]

最终确定结果以构建一个评分包。

Return type:

ScorePack