probly.calibration.scaling.common

Common methods for temperature scaling.

Functions

platt(base)

Dispatches to the correct platt scaling implementation.

register_platt_factory(key)

Returns a decorator to register a class in the platt factory.

register_temperature_factory(key)

Returns a decorator to register a class in the temperature factory.

register_vector_factory(key)

Returns a decorator to register a class in the vector factory.

temperature(base)

Dispatches to the correct temperature scaling implementation.

vector(base, num_classes)

Dispatches to the correct vector scaling implementation.

probly.calibration.scaling.common.platt(base)[source]

Dispatches to the correct platt scaling implementation.

Parameters:

base (object)

Return type:

object

probly.calibration.scaling.common.register_platt_factory(key)[source]

Returns a decorator to register a class in the platt factory.

Parameters:

key (LazyType)

Return type:

Callable

probly.calibration.scaling.common.register_temperature_factory(key)[source]

Returns a decorator to register a class in the temperature factory.

Parameters:

key (LazyType)

Return type:

Callable

probly.calibration.scaling.common.register_vector_factory(key)[source]

Returns a decorator to register a class in the vector factory.

Parameters:

key (LazyType)

Return type:

Callable

probly.calibration.scaling.common.temperature(base)[source]

Dispatches to the correct temperature scaling implementation.

Parameters:

base (object)

Return type:

object

probly.calibration.scaling.common.vector(base, num_classes)[source]

Dispatches to the correct vector scaling implementation.

Parameters:
Return type:

object