Skip to content

CrossEncoderOutput

class lightning_ir.cross_encoder.cross_encoder_model.CrossEncoderOutput(scores: Tensor | None = None, embeddings: Tensor | None = None)[source]

Bases: LightningIROutput

Dataclass containing the output of a cross-encoder model

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

Methods

__init__([scores, embeddings])

Args:

Attributes

Joint query-document embeddings

embeddings: Tensor | None = None: Joint query-document embeddings