pt-melt

Contents

  • pt-melt
    • PT-MELT package
      • Blocks Module
        • BayesianBlock
          • BayesianBlock.forward()
          • BayesianBlock.kl_loss()
        • DefaultOutput
          • DefaultOutput.forward()
        • DenseBlock
          • DenseBlock.forward()
        • MELTBlock
        • MixtureDensityOutput
          • MixtureDensityOutput.forward()
        • ResidualBlock
          • ResidualBlock.forward()
      • Layers Module
        • MELTBatchNorm
          • MELTBatchNorm.forward()
          • MELTBatchNorm.reset_parameters()
        • MELTBatchRenorm
          • MELTBatchRenorm.forward()
        • MELTBayesianDenseFlipOut
          • MELTBayesianDenseFlipOut.forward()
      • Losses Module
        • MixtureDensityLoss
          • MixtureDensityLoss.forward()
        • safe_exp()
      • Models Module
        • ArtificialNeuralNetwork
          • ArtificialNeuralNetwork.forward()
          • ArtificialNeuralNetwork.initialize_layers()
        • BayesianNeuralNetwork
          • BayesianNeuralNetwork.build()
          • BayesianNeuralNetwork.create_output_layer()
          • BayesianNeuralNetwork.forward()
          • BayesianNeuralNetwork.initialize_layers()
          • BayesianNeuralNetwork.step()
        • MELTModel
          • MELTModel.build()
          • MELTModel.compute_jacobian()
          • MELTModel.create_dropout_layers()
          • MELTModel.create_output_layer()
          • MELTModel.fit()
          • MELTModel.get_loss_fn()
          • MELTModel.initialize_layers()
          • MELTModel.l1_regularization()
          • MELTModel.l2_regularization()
          • MELTModel.step()
        • ResidualNeuralNetwork
          • ResidualNeuralNetwork.build()
          • ResidualNeuralNetwork.forward()
          • ResidualNeuralNetwork.initialize_layers()
      • NN Utils Module
        • get_activation()
        • get_initializer()
        • get_loss_fn()
      • Subpackages
        • PT-MELT Utilities Subpackage
          • Evaluation Module
          • Preprocessing Module
          • Statistics Module
          • Visualization Module
    • Example Notebooks
pt-melt
  • Search

© Copyright 2023, Alliance for Sustainable Energy.

Built with Sphinx using a theme provided by Read the Docs.