Update README.md

README.md

@@ -17,10 +17,9 @@ The model was trained with the [MNIST digits classification dataset](https://ker
 - Size: 1,256,080 parameters
 - Dataset: [tf.keras.datasets.mnist](https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz)
 - Language: English
-- Number of Training Epochs: 10
+- Number of Epochs: 10
 - Batch size: 32
-- Optimizer: Adam
-- Learning Rate: 0.001
+- Optimizer: Adam (tf.keras.optimizers.Adam(learning_rate=0.001)
 - Hardware: Tesla T4
 - This repository has the source [code used](https://github.com/Nkluge-correa/teeny-tiny_castle/blob/master/ML%20Intro%20Course/19_quick_AI_app.ipynb) to train this model.