kevinkyi commited on
Commit
8d4d7b8
·
verified ·
1 Parent(s): 71cc2a7

Add README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md CHANGED
@@ -53,6 +53,21 @@ We train a compact image classifier to predict whether an image **is a tomato (1
53
  "batch_size": 16
54
  }
55
  ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
56
  ## Limitations & Known Failure Modes
57
  - Extremely small dataset → risk of overfitting and unstable metrics.
58
  - Backgrounds and lighting variations can bias predictions.
 
53
  "batch_size": 16
54
  }
55
  ```
56
+
57
+ ## Results on Held out Test
58
+ accuracy: 0.83, f1: 0.80
59
+
60
+ ## Training curves and Early Stopping
61
+ Validation F1 was tracked each epoch with patience = 6. Training stopped once performance plateaued, preventing overfitting.
62
+
63
+ ## Reproducability
64
+ - Seed: 42
65
+ - Python: 3.12
66
+ - PyTorch: 2.4.1
67
+ - TorchVision: 0.19.1
68
+ - Optuna: 4.0.0
69
+ - Compute: Google Colab GPU (T4)
70
+
71
  ## Limitations & Known Failure Modes
72
  - Extremely small dataset → risk of overfitting and unstable metrics.
73
  - Backgrounds and lighting variations can bias predictions.