分类评估器

class ClassificationEvaluator(**kwargs)

基础类: Evaluator[ClassificationMetricKey]

使用分类指标的评估器。

初始化评估器。

Parameters:

kwargs – 传递给 Evaluator.__init__() 的基于关键字的参数。

方法总结

clear()	清除缓冲区和中间结果。
finalize()	计算最终结果，并清除缓冲区。
process_scores_(hrt_batch, target, scores[, ...])	处理一批三元组及其为所有实体计算的分数。

方法文档

clear() → None

清除缓冲区和中间结果。

Return type:

无

finalize() → ClassificationMetricResults

计算最终结果，并清除缓冲区。

Return type:

ClassificationMetricResults

process_scores_(hrt_batch: Tensor, target: Literal['head', 'relation', 'tail'], scores: Tensor, true_scores: Tensor | None = None, dense_positive_mask: Tensor | None = None) → None

处理一批三元组及其为所有实体计算的分数。

Parameters:

• hrt_batch (Tensor) – 形状: (batch_size, 3)

• target (Literal['head', 'relation', 'tail']) – 预测目标

• scores (Tensor) – 形状: (batch_size, num_entities)

• true_scores (Tensor | None) – 形状: (batch_size, 1)

• dense_positive_mask (Tensor | None) – 形状: (batch_size, num_entities) 一个可选的二进制（0/1）张量，表示其他真实实体。

Return type:

无