File size: 1,196 Bytes
1641932 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
license: mit
datasets:
- louiecerv/cats_dogs_dataset
language:
- en
pipeline_tag: image-classification
---
# Model Card for Cats vs Dogs Classifier
## Model Description
This model is a Convolutional Neural Network (CNN) designed to classify images of cats and dogs. It was trained on the "louiecerv/cats_dogs_dataset" dataset.
## Intended Use
The model is intended for educational purposes and as a demonstration of image classification using CNNs. It can be used to classify images of cats and dogs.
## Training Data
The model was trained on the "louiecerv/cats_dogs_dataset" dataset, which contains images of cats and dogs.
## Training Procedure
The model was trained using the following procedure:
- Optimizer: Adam
- Learning Rate: 0.001
- Loss Function: CrossEntropyLoss
- Number of Epochs: 10
- Batch Size: 32
## Evaluation
The model was evaluated on the training dataset. Further evaluation on a separate validation or test dataset is recommended.
## Limitations
The model may not perform well on images that are significantly different from the training data. It is also limited to binary classification (cats vs. dogs).
## License
The model is licensed under the MIT License. |