Skip to content

Commit

Permalink
Including DNN Extension
Browse files Browse the repository at this point in the history
  • Loading branch information
lorinanthony committed Jan 31, 2019
1 parent 67e7e4f commit 97de768
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Deep Learning/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

Here, we demonstrate how to implement RATE with Bayesian neural network architectures as described in [Ish-Horowicz et al. (2019)](https://arxiv.org/abs/1901.09839). The `Notebooks` directory contains notebooks used to generate the results and plots in the paper. These are meant to serve as examples of how to build and train Bayesian neural networks and determine variable importance for its input features.

The source code in `src` organized as follows:
The source code in the `src` directory is organized as follows:
* `BayesNN.py` contains a class implementing the Bayesian neural network.
* `rate_bnn.py` contains code for computing RATE values for a Bayesian neural network.
* `utils.py` contains functions for loading data and computing accuracies across Monte Carlo sampled predictions.

0 comments on commit 97de768

Please sign in to comment.