Skip to content

dtak/hs-bnn-public

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Horseshoe Priors for Bayesian Neural Networks.


This software includes an implementation of the model presented in:
"Model Selection in Bayesian Neural Networks via Horseshoe priors", Soumya Ghosh and Finale Doshi-Velez. NIPS 2017. Bayesian Deep Learning Workshop (http://bayesiandeeplearning.org/2017/papers/42.pdf).

The software depends on autograd (https://github.com/HIPS/autograd)

USAGE:

  • To train,
python ./scripts/train_hsbnn.py

  • You can use
python ./src/plot_weights.py path/to/results.pkl

to visualize the inferred weights.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages