|
--- |
|
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. |
