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