Skip to content

Commit bfed9c9

Browse files
committed
Add requirements.txt for BNN notebook
1 parent e75d565 commit bfed9c9

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

bayesian-neural-networks/bayesian_neural_networks.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
"source": [
77
"# Variational inference in Bayesian neural networks\n",
88
"\n",
9-
"This article demonstrates how to implement and train a Bayesian neural network with Keras following the approach described in [Weight Uncertainty in Neural Networks](https://arxiv.org/abs/1505.05424) (*Bayes by Backprop*). The implementation is kept simple for illustration purposes and uses Keras 2.2.4 and Tensorflow 1.12.0. For more advanced implementations of Bayesian methods for neural networks consider using [Tensorflow Probability](https://www.tensorflow.org/probability), for example.\n",
9+
"This article demonstrates how to implement and train a Bayesian neural network with Keras following the approach described in [Weight Uncertainty in Neural Networks](https://arxiv.org/abs/1505.05424) (*Bayes by Backprop*). The implementation is kept simple for illustration purposes and uses Keras 2.2.4 and Tensorflow 1.12.0 (see also [requirements.txt](requirements.txt)). For more advanced implementations of Bayesian methods for neural networks consider using [Tensorflow Probability](https://www.tensorflow.org/probability), for example.\n",
1010
"\n",
1111
"Bayesian neural networks differ from plain neural networks in that their weights are assigned a probability distribution instead of a single value or point estimate. These probability distributions describe the uncertainty in weights and can be used to estimate uncertainty in predictions. Training a Bayesian neural network via variational inference learns the parameters of these distributions instead of the weights directly.\n",
1212
"\n",
@@ -426,7 +426,7 @@
426426
"name": "python",
427427
"nbconvert_exporter": "python",
428428
"pygments_lexer": "ipython3",
429-
"version": "3.6.8"
429+
"version": "3.6.12"
430430
}
431431
},
432432
"nbformat": 4,
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
# Dependencies for bayesian_neural_networks.ipynb
2+
keras==2.2.4
3+
tensorflow==1.12
4+
tensorflow_probability==0.5.0
5+
matplotlib==3.3.2
6+
tqdm==4.50.2

0 commit comments

Comments
 (0)