This is a Bayesian Neural Network (BNN) implementation for
PyTorch. The implementation follows Yarin Gal's papers
"Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep
Learning" (see BDropout) and "Concrete Dropout" (see CDropout).
This package was originally based off the work here: juancamilog/prob_mbrl.
To install simply clone and run:
python setup.py installYou may also install the dependencies with pipenv as follows:
pipenv installFinally, you may add this to your own application with either:
pip install 'git+https://github.com/anassinator/bnn.git#egg=bnn'
pipenv install 'git+https://github.com/anassinator/bnn.git#egg=bnn'After installation, import and use as follows:
import bnnYou can see the examples directory for some Jupyter notebooks with more detailed examples.
The following is an example of what this BNN was able to estimate
with a few randomly sampled points (in red) of a noisy sin function.
The dotted curve represent the real function that was kept a secret from the
model, whereas the black line and the grey area represent the
estimated mean and uncertainty.
Contributions are welcome. Simply open an issue or pull request on the matter.
We use YAPF for all Python formatting needs. You can auto-format your changes with the following command:
yapf --recursive --in-place --parallel .You can install the formatter with:
pipenv install --devSee LICENSE.