mixmo.core.temperature_scaling

Temperature scaling functions and networks modules Taken from https://github.com/gpleiss/temperature_scaling/blob/master/temperature_scaling.py

Functions

apply_temperature_on_logits(logits, temperature)

Apply temperature relaxation on logits

Classes

NetworkWithTemperature(network[, …])

A thin decorator, which wraps a network with temperature scaling network (nn.Module):