File size: 1,223 Bytes
cd44324 b22cc92 cd44324 64a8f1e b22cc92 cd44324 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
---
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
datasets:
- garythung/trashnet
pipeline_tag: image-classification
---
- 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](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
- Library: [More Information Needed]
- Docs: [More Information Needed] |