Skip to content

laplax.enums

Enums for Laplace approximations.

CalibrationObjective

Bases: StrEnum

Supported calibration objectives (minimisation!).

Source code in laplax/enums.py 
35
36
37
38
39
40
41
class CalibrationObjective(StrEnum):
    """Supported calibration objectives (minimisation!)."""

    NLL = "nll"
    CHI_SQUARED = "chi_squared"
    MARGINAL_LOG_LIKELIHOOD = "marginal_log_likelihood"
    ECE = "ece"

CalibrationMethod

Bases: StrEnum

Supported calibration methods.

Source code in laplax/enums.py 
44
45
46
47
class CalibrationMethod(StrEnum):
    """Supported calibration methods."""

    GRID_SEARCH = "grid_search"

Pushforward

Bases: StrEnum

Supported pushforward types for pushing forward weight space uncertainty.

Source code in laplax/enums.py 
50
51
52
53
54
class Pushforward(StrEnum):
    """Supported pushforward types for pushing forward weight space uncertainty."""

    LINEAR = "linear"
    NONLINEAR = "nonlinear"

Predictive

Bases: StrEnum

Supported predictive types for crossing softmax transformation.

Source code in laplax/enums.py 
57
58
59
60
61
62
63
64
65
class Predictive(StrEnum):
    """Supported predictive types for crossing softmax transformation."""

    MC_BRIDGE = "mc_bridge"
    LAPLACE_BRIDGE = "laplace_bridge"
    MEAN_FIELD_0 = "mean_field_0"
    MEAN_FIELD_1 = "mean_field_1"
    MEAN_FIELD_2 = "mean_field_2"
    NONE = "none"  # Intended for regression flag

DefaultMetrics

Bases: StrEnum

Supported default metric settings.

Source code in laplax/enums.py 
68
69
70
71
72
class DefaultMetrics(StrEnum):
    """Supported default metric settings."""

    CLASSIFICATION = "classification"
    REGRESSION = "regression"