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