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.8722
+**Validation Accuracy:** 0.8647
 
 ## Training Configuration
 

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-Healthy,1.0,0.7045454545454546,0.8266666666666667,44.0
-Bean Rust,0.7258064516129032,1.0,0.8411214953271028,45.0
-Angular Leaf Spot,1.0,0.9090909090909091,0.9523809523809523,44.0
-accuracy,0.8721804511278195,0.8721804511278195,0.8721804511278195,0.8721804511278195
-macro avg,0.9086021505376344,0.8712121212121212,0.873389704791574,133.0
-weighted avg,0.9072277467863206,0.8721804511278195,0.8731470866752997,133.0
+Healthy,1.0,0.6590909090909091,0.7945205479452054,44.0
+Bean Rust,0.7142857142857143,1.0,0.8333333333333334,45.0
+Angular Leaf Spot,1.0,0.9318181818181818,0.9647058823529412,44.0
+accuracy,0.8646616541353384,0.8646616541353384,0.8646616541353384,0.8646616541353384
+macro avg,0.9047619047619048,0.8636363636363636,0.86418658787716,133.0
+weighted avg,0.9033297529538131,0.8646616541353384,0.8639546085196875,133.0

resnet18_beans.pth

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