CountScoreConsumer

class CountScoreConsumer[源代码]

一个简单的消费者，用于计算批次和分数的数量。

初始化消费者。

方法总结

__call__(batch, target, scores)

消耗给定 hr_batch 的分数。

方法文档

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

消耗给定 hr_batch 的分数。

Parameters:

Return type:

无