Downloads · 30 days
0
autogluon-image-predictor
yl0628/autogluon-image-predictor
autogluon-image-predictor is a machine learning model from yl0628. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
- Framework: AutoGluon - Task: Classification
Access
Public
Updated Sep 22, 2025
Repo size
86.5 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pkl44.9 MB · 52%
At a glance
- License
- mit
- Access
- Public
- Created
- Sep 22, 2025
- Updated
- Sep 22, 2025
- SHA
- c55f8b4f
Datasets
Spaces
Model card
From the Hugging Face model README
Model Card: AutoML Neural Network Predictor for Tomato Images
Model Details
- Framework:
AutoGluon - Task:
Classification
Dataset
- Source: Iris314/Food_tomatoes_dataset
- Target:
label - Splits:
- Augmented: 490 rows
- Original: 49 rows
- Preprocessing Steps:
- Stratify 'label' column.
- Train/test split (80%/20%).
Model
| Name | Type | Params | Mode |
|---|---|---|---|
| model | TimmAutoModelForImagePrediction | 11.2 M | train |
| validation_metric | MulticlassAccuracy | 0 | train |
| loss_func | CrossEntropyLoss | 0 | train |
Summary
- Trainable params: 11.2 M
- Non-trainable params: 0
- Total params: 11.2 M
- Total estimated model params size: 44.710 MB
- Modules in train mode: 101
- Modules in eval mode: 0
- Validation accuracy: 1
- Training time: ~49.5 seconds
Training
- Framework: AutoGluon
- Preset:
"medium_quality" - Image Size: 224x224
- Explored Models: ResNet 18
Results
- Test Split:
- Accuracy: 0.9796
- Weighted F1: 0.9796
Notes
Educational use only. Used AutoML for training model, used ChatGPT and Gemini to debug, used ChatGPT to make table for model info.
- License
- mit
- Languages
- en
- Created
- Sep 22, 2025
- Updated
- Sep 22, 2025