RonMcKay / bnn

PyTorch Bayesian Neural Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

An implementation of bayesian neural networks which I use for research purposes.

Installation

pip install git+https://github.com/RonMcKay/bnn

The following repositories were partly a basis for this package:

Javier Antoran - Bayesian Neural Networks
Kumar Shridhar - PyTorch Bayesian CNN

Development

If you want to work on this repository, clone it (or your fork)

git clone https://github.com/RonMcKay/bnn.git

and install it using poetry

poetry install

Please make all your pull-requests against dev and ask in the discussion section before you go about making a change. Also make sure that all formatting tests are passing for you by running make test.

About

PyTorch Bayesian Neural Networks

License:MIT License


Languages

Language:Python 99.6%Language:Makefile 0.4%