Skip to content

Commit 97de768

Browse files
committed
Including DNN Extension
1 parent 67e7e4f commit 97de768

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Deep Learning/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
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.
44

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

0 commit comments

Comments
 (0)