primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput#

class primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput(loss: Optional[torch.FloatTensor] = None, start_logits: Optional[torch.FloatTensor] = None, end_logits: Optional[torch.FloatTensor] = None, target_type_logits: Optional[torch.FloatTensor] = None, hidden_states: Optional[Tuple[torch.FloatTensor]] = None, attentions: Optional[Tuple[torch.FloatTensor]] = None)#

Bases: transformers.file_utils.ModelOutput

Extractive QA model outputs comprising: (loss), start_logits, end_logits, target_type_logits, (hidden_states), (attentions).

Methods

clear

copy

fromkeys

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

get

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

items

keys

move_to_end

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

pop

value.

popitem

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

setdefault

Insert key with a value of default if key is not in the dictionary.

to_tuple

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

update

If E is present and has a .keys() method, then does: for k in E: D[k] = E[k] If E is present and lacks a .keys() method, then does: for k, v in E: D[k] = v In either case, this is followed by: for k in F: D[k] = F[k]

values

Attributes

attentions

end_logits

hidden_states

loss

start_logits

target_type_logits

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.

items() a set-like object providing a view on D's items#
keys() a set-like object providing a view on D's keys#
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.

pop(k[, d]) v, remove specified key and return the corresponding#

value. If key is not found, d is returned if given, otherwise KeyError is raised.

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.

setdefault(*args, **kwargs)#

Insert key with a value of default if key is not in the dictionary.

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

to_tuple() Tuple[Any]#

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

update([E, ]**F) None.  Update D from dict/iterable E and F.#

If E is present and has a .keys() method, then does: for k in E: D[k] = E[k] If E is present and lacks a .keys() method, then does: for k, v in E: D[k] = v In either case, this is followed by: for k in F: D[k] = F[k]

values() an object providing a view on D's values#