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
    • AttentionPool
      • AttentionPool.forward()
    • 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.get_optimizer()
      • MELTModel.get_scheduler()
      • MELTModel.initialize_layers()
      • MELTModel.l1_regularization()
      • MELTModel.l2_regularization()
      • MELTModel.step()
    • RecurrentNeuralNetwork
      • RecurrentNeuralNetwork.create_output_layer()
      • RecurrentNeuralNetwork.forward()
      • RecurrentNeuralNetwork.initialize_layers()
      • RecurrentNeuralNetwork.step()
    • ResidualNeuralNetwork
      • ResidualNeuralNetwork.build()
      • ResidualNeuralNetwork.forward()
      • ResidualNeuralNetwork.initialize_layers()
  • NN Utils Module
  • Subpackages
    • PT-MELT Utilities Subpackage
      • Evaluation Module
      • Preprocessing Module
      • Statistics Module
      • Visualization Module
• Example Notebooks