Downloads · 30 days
0
ai-image-portofolio
Kx15/ai-image-portofolio
ai-image-portofolio is a machine learning model from Kx15. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is a fine-tuned version of torchvision.models.wideresnet502 on the indonesian-food-dataset. It achieves the following results on the evaluation set: - Loss: 0.602 - Accuracy: 0.6704
Access
Public
Updated Oct 18, 2023
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
Other1.5 KB · 58%
At a glance
- Access
- Public
- Created
- Oct 18, 2023
- Updated
- Oct 18, 2023
- SHA
- d1328724
Model card
From the Hugging Face model README
indonesian food_classification
This model is a fine-tuned version of torchvision.models.wide_resnet50_2 on the indonesian-food-dataset. It achieves the following results on the evaluation set:
- Loss: 0.602
- Accuracy: 0.6704
Model description
This model is based on the torchvision.models.wide_resnet50_2 architecture, which is a pre-trained model. The model has fine-tuned for a specific task using the Indonesian Food Dataset.
Intended uses & limitations
Intended Uses: This model is intended for image classification tasks, specifically for classifying Indonesian food items in the Indonesian Food Dataset. Limitations: Deep learning models require a large amount of data for training. They might not perform well if the dataset is small or not representative
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.01
- train_batch_size: 50
- eval_batch_size: 32
- optimizer: SGD
- lr_scheduler_type: MultiStepLr
- num_epochs: 100
Training results
Best Accuracy Achieved: 0.6704
- Languages
- en
- Created
- Oct 18, 2023
- Updated
- Oct 18, 2023