Skip to content

LightningIROutput

class lightning_ir.base.model.LightningIROutput(scores: Tensor | None = None)[source]

Bases: ModelOutput

Base class for the output of the Lightning IR model. It is a subclass of transformers.ModelOutput.

scores

Output relevance scores for query–document pairs. Defaults to None.

Type:: torch.Tensor | None

__init__(scores: Tensor | None = None) → None: Args:

Methods

__init__([scores])

Args:

Attributes