class documentation

Mixin for Model classes that can generate logits for their predictions.

This mixin allows for the possibility that only some instances of a class are capable of generating logits, per the value of the has_logits property.

Method __init__ Undocumented
Method store_logits.setter Undocumented
Property has_logits Whether this instance can generate logits.
Property store_logits Whether the model should store logits in its predictions.
Instance Variable _store_logits Undocumented
@store_logits.setter
def store_logits(self, flag): (source)

Undocumented

Whether this instance can generate logits.

@property
store_logits = (source)

Whether the model should store logits in its predictions.

_store_logits = (source)

Undocumented