Merge branch 'main' of https://huggingface.co/brendenc/ProbabalisticBayesianModel-Wine

README.md

@@ -7,23 +7,22 @@ tags:
 
 ## Model description
 
-More information needed
+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.
 
-## Intended uses & limitations
+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.
 
-More information needed
+**Full credits go to [Khalid Salama](https://www.linkedin.com/in/khalid-salama-24403144/)**
 
-## Training and evaluation data
+## Training and evaluation data 🍷
 
-More information needed
+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.
 
-## Training procedure
+## Versioning
+
+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.
 
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
 - optimizer: {'name': 'RMSprop', 'learning_rate': 0.001, 'decay': 0.0, 'rho': 0.9, 'momentum': 0.0, 'epsilon': 1e-07, 'centered': False}
 - training_precision: float32
-
- ## Training Metrics
-Model history needed