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

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-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
+Healthy,1.0,0.75,0.8571428571428571,44.0
+Bean Rust,0.75,1.0,0.8571428571428571,45.0
+Angular Leaf Spot,1.0,0.9090909090909091,0.9523809523809523,44.0
+accuracy,0.8872180451127819,0.8872180451127819,0.8872180451127819,0.8872180451127819
+macro avg,0.9166666666666666,0.8863636363636364,0.8888888888888888,133.0
+weighted avg,0.9154135338345865,0.8872180451127819,0.8886501969208735,133.0

resnet18_beans.pth

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