| language: | |
| - en | |
| license: mit # change if you prefer a different license | |
| library_name: pytorch | |
| tags: | |
| - vision | |
| - computer-vision | |
| - image-classification | |
| - automl | |
| - neural-network | |
| - early-stopping | |
| - colab | |
| - pytorch | |
| datasets: | |
| - ysakhale/gym-machines-image-dataset | |
| metrics: | |
| - accuracy | |
| - f1 | |
| - precision | |
| - recall | |
| - loss | |