brendenc commited on
Commit
42ee490
1 Parent(s): fbfc3b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -11,6 +11,12 @@ This repo contains model weights for the the probabilistic model from [Probabili
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.
 
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
+ **Full credits go to [Khalid Salama](https://www.linkedin.com/in/khalid-salama-24403144/)**
15
+
16
+ ## Training and evaluation data 🍷
17
+
18
+ We use the wine quality dataset found [here](https://www.tensorflow.org/datasets/catalog/wine_quality). Each wine was scored from 0-10 by wine experts, and includes 11 physicochemical features about the wine.
19
+
20
  ## Versioning
21
 
22
  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.