CrossEncoderOutput

class lightning_ir.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

Methods

__init__([scores, embeddings])

clear()

copy()

fromkeys(iterable[, value])

Create a new ordered dictionary with keys from iterable and values set to value.

get(key[, default])

Return the value for key if key is in the dictionary, else default.

items

keys

move_to_end(key[, last])

Move an existing element to the end (or beginning if last is false).

pop(*args, **kwargs)

popitem([last])

Remove and return a (key, value) pair from the dictionary.

setdefault(*args, **kwargs)

to_tuple()

Convert self to a tuple containing all the attributes/keys that are not None.

update(*args, **kwargs)

values

Attributes

embeddings

Joint query-document embeddings

scores

clear() → None.  Remove all items from od.
copy() → a shallow copy of od
embeddings: Tensor | None = None

Joint query-document embeddings

classmethod fromkeys(iterable, value=None)

Create a new ordered dictionary with keys from iterable and values set to value.

get(key, default=None, /)

Return the value for key if key is in the dictionary, else default.

move_to_end(key, last=True)

Move an existing element to the end (or beginning if last is false).

Raise KeyError if the element does not exist.

popitem(last=True)

Remove and return a (key, value) pair from the dictionary.

Pairs are returned in LIFO order if last is true or FIFO order if false.

to_tuple() → Tuple[Any]

Convert self to a tuple containing all the attributes/keys that are not None.