climb.tool.impl.data_suite.third_party.uq360.models package

Subpackages

• climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks package
  • Subpackages
    • climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks.bnn_models package
      • Submodules
      • climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks.bnn_models.bayesian_mlp module
      • climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks.bnn_models.horseshoe_mlp module
      • Module contents
  • Submodules
  • climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks.layer_utils module
    • InvGammaHalfCauchyLayer
      • InvGammaHalfCauchyLayer.entropy()
      • InvGammaHalfCauchyLayer.entropy_a2()
      • InvGammaHalfCauchyLayer.entropy_lambda()
      • InvGammaHalfCauchyLayer.expectation_wrt_prior()
      • InvGammaHalfCauchyLayer.fixed_point_updates()
      • InvGammaHalfCauchyLayer.initialize_from_prior()
      • InvGammaHalfCauchyLayer.kl()
    • InvGammaLayer
      • InvGammaLayer.entropy()
      • InvGammaLayer.expectation_wrt_prior()
      • InvGammaLayer.initialize_from_prior()
      • InvGammaLayer.kl()
    • diag_gaussian_entropy()
    • inv_gamma_entropy()
    • log_normal_entropy()
  • climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks.layers module
    • BayesianLinearLayer
      • BayesianLinearLayer.forward()
      • BayesianLinearLayer.init_parameters()
      • BayesianLinearLayer.kl()
    • HorseshoeLayer
      • HorseshoeLayer.fixed_point_updates()
      • HorseshoeLayer.forward()
      • HorseshoeLayer.kl()
    • NodeSpecificRegularizedHorseshoeLayer
    • RegularizedHorseshoeLayer
      • RegularizedHorseshoeLayer.forward()
      • RegularizedHorseshoeLayer.kl()
    • reparam()
  • climb.tool.impl.data_suite.third_party.uq360.models.bayesian_neural_networks.misc module
    • compute_test_ll()
  • Module contents
• climb.tool.impl.data_suite.third_party.uq360.models.noise_models package
  • Submodules
  • climb.tool.impl.data_suite.third_party.uq360.models.noise_models.heteroscedastic_noise_models module
    • GaussianNoise
      • GaussianNoise.get_noise_var()
      • GaussianNoise.loss()
    • transform()
  • climb.tool.impl.data_suite.third_party.uq360.models.noise_models.homoscedastic_noise_models module
    • GaussianNoiseFixedPrecision
      • GaussianNoiseFixedPrecision.get_noise_var()
      • GaussianNoiseFixedPrecision.loss()
    • GaussianNoiseGammaPrecision
      • GaussianNoiseGammaPrecision.get_noise_var()
      • GaussianNoiseGammaPrecision.kl()
      • GaussianNoiseGammaPrecision.loss()
    • transform()
  • climb.tool.impl.data_suite.third_party.uq360.models.noise_models.noisemodel module
    • AbstractNoiseModel
      • AbstractNoiseModel.get_noise_var()
      • AbstractNoiseModel.loss()
  • Module contents

Submodules

climb.tool.impl.data_suite.third_party.uq360.models.heteroscedastic_mlp module

class climb.tool.impl.data_suite.third_party.uq360.models.heteroscedastic_mlp.GaussianNoiseMLPNet(num_features, num_outputs, num_hidden)

Bases: Module

forward(x)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

loss(y_true=None, mu_pred=None, log_var_pred=None)

Module contents