Downloads · 30 days
9
4% of all-time downloads
trash-clasification
pradanaadn/trash-clasification
trash-clasification is a image classification model from pradanaadn. Use it when you need a label for an image.
- Epoch: 10 - Optimizer: Adam - Learning Rate: 10e-3 - Augmentation: Random Horizontal and Vertical Flip
All-time downloads
208
Public
Parameters
4.2M
1.9 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors17 MB · 100%
At a glance
- Task
- Image Classification
- Model type
- mobilenet_v3_large
- Access
- Public
- Created
- Dec 21, 2024
- Updated
- Dec 22, 2024
- SHA
- e0b9403d
Datasets
Model card
From the Hugging Face model README
- Epoch: 10
- Optimizer: Adam
- Learning Rate: 10e-3
- Augmentation: Random Horizontal and Vertical Flip
Result:
| Metric | Train | Validation | Test |
|---|---|---|---|
| Loss | 0.29818 | 0.34164 | 0.3223 |
| Accuracy (Macro) | 0.89413 | 0.87473 | 0.86351 |
| Accuracy (Micro) | 0.90601 | 0.88515 | 0.88538 |
| Accuracy (Weighted) | 0.90601 | 0.88515 | 0.88538 |
| F1 (Macro) | 0.89576 | 0.87189 | 0.8632 |
| F1 (Micro) | 0.90601 | 0.88515 | 0.88538 |
| F1 (Weighted) | 0.90581 | 0.88571 | 0.88556 |
| Precision (Macro) | 0.89771 | 0.87115 | 0.86314 |
| Precision (Micro) | 0.90601 | 0.88515 | 0.88538 |
| Precision (Weighted) | 0.9059 | 0.88785 | 0.88599 |
| Recall (Macro) | 0.89413 | 0.87473 | 0.86351 |
| Recall (Micro) | 0.90601 | 0.88515 | 0.88538 |
| Recall (Weighted) | 0.90601 | 0.88515 | 0.88538 |
This model has been pushed to the Hub using the PytorchModelHubMixin integration:
- Library: [More Information Needed]
- Docs: [More Information Needed]
- Task
- Image Classification
- Type
- mobilenet_v3_large
- Created
- Dec 21, 2024
- Updated
- Dec 22, 2024