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.7895
+**Validation Accuracy:** 0.8947
 
 ## 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-06
+- Learning Rate: 5e-05
 - Epochs: 5
 - Batch Size: 32
 - Weight Decay: 0.0001
-- Dropout: 0.2
+- Dropout: 0.25
 - Optimizer: Adam
 
 ## Artifacts
@@ -77,6 +77,5 @@ Download and load the model:
     model.load_state_dict(torch.load(model_path, map_location="cpu"))
     model.eval()
 
-## License
 
 Apache 2.0

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-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
+Healthy,1.0,0.7272727272727273,0.8421052631578947,44.0
+Bean Rust,0.7627118644067796,1.0,0.8653846153846154,45.0
+Angular Leaf Spot,1.0,0.9545454545454546,0.9767441860465116,44.0
+accuracy,0.8947368421052632,0.8947368421052632,0.8947368421052632,0.8947368421052632
+macro avg,0.9209039548022598,0.8939393939393939,0.8947446881963406,133.0
+weighted avg,0.91971454058876,0.8947368421052632,0.8945239357691847,133.0

resnet18_beans.pth

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