xpectra.TemperatureScaler

class xpectra.TemperatureScaler[source]

Bases: object

Single-parameter temperature scaling fitted by NLL minimization.

__init__()[source]

Return type:: None

Methods

`__init__`()
`fit`(logits, y)
`transform`(logits)

fit(logits, y)[source]

Parameters:

logits (ndarray)
y (ndarray)

Return type:

TemperatureScaler

transform(logits)[source]

Parameters:: logits (ndarray)
Return type:: ndarray