Update model with improved regularization and data augmentation

README.md

@@ -30,7 +30,7 @@ This model was trained in Google Colab using a T4 GPU and tracked with MLflow.
 - Bean Rust
 - Angular Leaf Spot
 
-**Validation Accuracy:** 0.9023
+**Validation Accuracy:** 0.7895
 
 ## Training Configuration
 
@@ -42,11 +42,11 @@ This model was trained in Google Colab using a T4 GPU and tracked with MLflow.
 - Best model selection based on validation accuracy
 
 **Hyperparameters:**
-- Learning Rate: 5e-05
+- Learning Rate: 5e-06
 - Epochs: 5
 - Batch Size: 32
 - Weight Decay: 0.0001
-- Dropout: 0.3
+- Dropout: 0.2
 - Optimizer: Adam
 
 ## Artifacts

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-Healthy,1.0,0.7727272727272727,0.8717948717948718,44.0
-Bean Rust,0.7758620689655172,1.0,0.8737864077669902,45.0
-Angular Leaf Spot,1.0,0.9318181818181818,0.9647058823529412,44.0
-accuracy,0.9022556390977443,0.9022556390977443,0.9022556390977443,0.9022556390977443
-macro avg,0.9252873563218391,0.9015151515151515,0.9034290539716011,133.0
-weighted avg,0.9241638579206638,0.9022556390977443,0.9032061769324686,133.0
+Healthy,0.8846153846153846,0.5227272727272727,0.6571428571428571,44.0
+Bean Rust,0.6323529411764706,0.9555555555555556,0.7610619469026548,45.0
+Angular Leaf Spot,1.0,0.8863636363636364,0.9397590361445783,44.0
+accuracy,0.7894736842105263,0.7894736842105263,0.7894736842105263,0.7894736842105263
+macro avg,0.8389894419306184,0.7882154882154881,0.78598794673003,133.0
+weighted avg,0.8374357840302112,0.7894736842105263,0.7858005331974934,133.0

resnet18_beans.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:51d514069813572bd7c242f17108e7981478631b25cd00b9679bea74f35af460
+oid sha256:8d80a5067fc28b6b7c2fb185806085e26ecdcf2db04b680b3b3d6cfc2dbce54d
 size 44792395