|
--- |
|
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] |
