ScoreConsumer

class ScoreConsumer

Bases: object

A consumer of scores for visitor pattern.

Methods Summary

__call__(batch, target, scores)	Consume scores for the given hr_batch.
finalize()	Finalize the result to build a score pack.

Methods Documentation

abstract __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

finalize() → ScorePack

Finalize the result to build a score pack.

Return type:

ScorePack