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.9098
+**Validation Accuracy:** 0.9398
 
 ## Training Configuration
 
@@ -46,7 +46,7 @@ This model was trained in Google Colab using a T4 GPU and tracked with MLflow.
 - Epochs: 10
 - Batch Size: 32
 - Weight Decay: 0.0001
-- Dropout: 0.35
+- Dropout: 0.3
 - Optimizer: Adam
 
 ## Artifacts

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-Healthy,0.972972972972973,0.8181818181818182,0.8888888888888888,44.0
-Bean Rust,0.8113207547169812,0.9555555555555556,0.8775510204081632,45.0
-Angular Leaf Spot,0.9767441860465116,0.9545454545454546,0.9655172413793104,44.0
-accuracy,0.9097744360902256,0.9097744360902256,0.9097744360902256,0.9097744360902256
-macro avg,0.9203459712454887,0.9094276094276094,0.9106523835587875,133.0
-weighted avg,0.9195262327753495,0.9097744360902256,0.9104035011290835,133.0
+Healthy,1.0,0.8863636363636364,0.9397590361445783,44.0
+Bean Rust,0.8490566037735849,1.0,0.9183673469387755,45.0
+Angular Leaf Spot,1.0,0.9318181818181818,0.9647058823529412,44.0
+accuracy,0.9398496240601504,0.9398496240601504,0.9398496240601504,0.9398496240601504
+macro avg,0.949685534591195,0.9393939393939393,0.9409440884787651,133.0
+weighted avg,0.948928926088807,0.9398496240601504,0.940774338542374,133.0

resnet18_beans.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e66ac4203a4dd03b003e8ed6984d475f764993e0aff003f2fb495afe4cae36dd
+oid sha256:bdaaee60924b2d69d08666ce33f3b600e5a025a2cb41588aa24ec7b52127928f
 size 44792395