LightningIROutput

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

Bases: ModelOutput

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

Parameters:

scores (torch.Tensor | None, optional) – Output relevance scores for query–document pairs, defaults to None

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

Methods

__init__([scores])

clear()

copy()

fromkeys([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

scores

clear() → None.  Remove all items from od.
copy() → a shallow copy of od
fromkeys(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.