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

# Files in this Repository
- `autogluon_image_predictor.pkl`: The trained `MultiModalPredictor` pickled using `cloudpickle`.
- `autogluon_image_predictor_dir.zip`: The zipped native AutoGluon predictor directory for portability.

# Communication
Feel free to contact me for any questions: aslann@cmu.edu