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

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-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
+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

resnet18_beans.pth

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