Add README.md
Browse files
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.
|