Update README.md

README.md

@@ -30,13 +30,14 @@ Model:
 4. Trainable Params: 136,839
 5. Accuracy: 0.823 | Precision: 0.825 | Recall: 0.823 | F1: 0.821
 
-Room for Improvement:
+## Room for Improvement:
 
 This model was created with extremely limited hardware acceleration (GPU) resources. Therefore, it is high likely that evaluation metrics that surpass the 95% mark can be achieved in the following manner:
 
 1. MobileNetv2 was used for its fast inference and low latency but perhaps, with more resources, a more suitable base model can be found.
 2. Data augmentation in order to better correct for class imbalances.
-3. Using a learning rate scheduler to train for longer (with lower LR) after nearing local minima (aprox 60 epochs).
+3. Using learning rate decay to train for longer (with lower LR) after nearing local minima (aprox 60 epochs).
+4. Error Analysis
 
 
 ## Uses
@@ -119,9 +120,8 @@ Use the code below to get started with the model locally:
         main()
 
 
-### Training Data
 
-Dataset used: FER (available on Kaggle)
+
 
 #### Preprocessing [optional]