Skip to content

Latest commit

 

History

History
53 lines (31 loc) · 2.53 KB

README.md

File metadata and controls

53 lines (31 loc) · 2.53 KB

Bayesian-Neural-Network-Pytorch

MIT License Pypi Documentation Status

This is a lightweight repository of bayesian neural network for PyTorch.

Usage

📋 Dependencies

  • torch 1.2.0
  • python 3.6

🔨 Installation

  • pip install torchbnn or
  • git clone https://github.com/Harry24k/bayesian-neural-network-pytorch
import torchbnn

🔎 Update Records

Here is update records of this package.

🚀 Demos

  • Bayesian Neural Network Regression (code): In this demo, two-layer bayesian neural network is constructed and trained on simple custom data. It shows how bayesian-neural-network works and randomness of the model.
  • Bayesian Neural Network Classification (code): To classify Iris data, in this demo, two-layer bayesian neural network is constructed and trained on the Iris data. It shows how bayesian-neural-network works and randomness of the model.
  • Convert to Bayesian Neural Network (code): To convert a basic neural network to a bayesian neural network, this demo shows how nonbayes_to_bayes and bayes_to_nonbayes work.
  • Freeze Bayesian Neural Network (code): To freeze a bayesian neural network, which means force a bayesian neural network to output same result for same input, this demo shows the effect of freeze and unfreeze.

Thanks to