louiecerv's picture
Update README.md
1641932 verified
metadata
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.