README.md

metadata

language:
  - en
license: mit
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: [email protected]