brendenc commited on
Commit
fbfc3b9
1 Parent(s): 49ca38a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -11
README.md CHANGED
@@ -7,23 +7,16 @@ tags:
7
 
8
  ## Model description
9
 
10
- More information needed
11
 
12
- ## Intended uses & limitations
13
 
14
- More information needed
15
 
16
- ## Training and evaluation data
17
-
18
- More information needed
19
-
20
- ## Training procedure
21
 
22
  ### Training hyperparameters
23
 
24
  The following hyperparameters were used during training:
25
  - optimizer: {'name': 'RMSprop', 'learning_rate': 0.001, 'decay': 0.0, 'rho': 0.9, 'momentum': 0.0, 'epsilon': 1e-07, 'centered': False}
26
  - training_precision: float32
27
-
28
- ## Training Metrics
29
- Model history needed
 
7
 
8
  ## Model description
9
 
10
+ This repo contains model weights for the the probabilistic model from [Probabilistic Bayesian Neural Networks](https://keras.io/examples/keras_recipes/bayesian_neural_networks/). This example demonstrates how to build basic probabilistic Bayesian neural networks to account for these two types of uncertainty. We use TensorFlow Probability library, which is compatible with Keras API.
11
 
12
+ Taking a probabilistic approach to deep learning allows to account for uncertainty, so that models can assign less levels of confidence to incorrect predictions. Sources of uncertainty can be found in the data, due to measurement error or noise in the labels, or the model, due to insufficient data availability for the model to learn effectively.
13
 
14
+ ## Versioning
15
 
16
+ The training was done using TensorFlow 2.8.0 and TensorFlow Probability 0.16.0. When working with TensorFlow Probability, it is encouraged to check out the [releases](https://github.com/tensorflow/probability/releases/tag/v0.17.0) to make sure you are using a stable TensorFlow counterpart.
 
 
 
 
17
 
18
  ### Training hyperparameters
19
 
20
  The following hyperparameters were used during training:
21
  - optimizer: {'name': 'RMSprop', 'learning_rate': 0.001, 'decay': 0.0, 'rho': 0.9, 'momentum': 0.0, 'epsilon': 1e-07, 'centered': False}
22
  - training_precision: float32