AllScoreConsumer

class AllScoreConsumer(num_entities: int, num_relations: int)

基础类：ScoreConsumer

收集所有三元组的分数。

初始化消费者。

Parameters:

• num_entities (int) – 实体的数量

• num_relations (int) – 关系的数量

属性摘要

flatten

方法总结

__call__(batch, target, scores)

消耗给定 hr_batch 的分数。

属性文档

flatten: bool = True

方法文档

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

消耗给定 hr_batch 的分数。

Parameters:

• batch (Tensor)

• 目标 (Literal['head', 'relation', 'tail'])

• 分数 (Tensor)

Return type:

无