CountScoreConsumer

class CountScoreConsumer

Bases: ScoreConsumer

A simple consumer which counts the number of batches and scores.

Initialize the consumer.

Methods Summary

__call__(batch, target, scores)

Consume scores for the given hr_batch.

Methods Documentation

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

Consume scores for the given hr_batch.

Parameters:

• batch (Tensor)

• target (Literal['head', 'relation', 'tail'])

• scores (Tensor)

Return type:

None