Downloads · 30 days
0
AnimalFacesCNN
michellewidjaja/AnimalFacesCNN
AnimalFacesCNN is a machine learning model from michellewidjaja. 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 CNN trained on the Kaggle Animal Face dataset to classify cats, dogs, and wildlife. It is based on the ResNet18 architecture.
Access
Public
Updated Sep 21, 2024
Repo size
44.8 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth44.8 MB · 100%
At a glance
- Access
- Public
- Created
- Sep 20, 2024
- Updated
- Sep 21, 2024
- SHA
- 0316448e
Model card
From the Hugging Face model README
Animal Faces CNN Model
This model is a CNN trained on the Kaggle Animal Face dataset to classify cats, dogs, and wildlife. It is based on the ResNet18 architecture.
Training Results:
- Validation Accuracy: 99.32%
- Dataset: Cats, Dogs, and Wildlife Faces
- Number of Epochs: 10
| Epoch | Loss | Validation Accuracy |
|---|---|---|
| 1 | 0.2374 | 95.11% |
| 2 | 0.1716 | 95.86% |
| 3 | 0.1381 | 97.53% |
| 4 | 0.1250 | 97.48% |
| 5 | 0.1038 | 97.53% |
| 6 | 0.0867 | 97.61% |
| 7 | 0.0914 | 98.44% |
| 8 | 0.0656 | 99.09% |
| 9 | 0.0462 | 99.35% |
| 10 | 0.0432 | 99.32% |
- Created
- Sep 20, 2024
- Updated
- Sep 21, 2024