CoilOutput

class lightning_ir.models.bi_encoders.coil.CoilOutput(scores: Tensor | None = None, query_embeddings: CoilEmbedding | None = None, doc_embeddings: CoilEmbedding | None = None, similarity: Tensor | None = None)[source]

Bases: BiEncoderOutput

Dataclass containing the output of a COIL model.

__init__(scores: Tensor | None = None, query_embeddings: CoilEmbedding | None = None, doc_embeddings: CoilEmbedding | None = None, similarity: Tensor | None = None) → None: Args:

Methods

__init__([scores, query_embeddings, ...])

Args:

Attributes

`doc_embeddings`	Document embeddings generated by the model.
`query_embeddings`	Query embeddings generated by the model.

doc_embeddings: CoilEmbedding | None = None: Document embeddings generated by the model.

query_embeddings: CoilEmbedding | None = None: Query embeddings generated by the model.