---
library_name: keras
tags:
- Image Classification
language:
- en
metrics:
- accuracy
---

# cats-vs-dogs

## Model Overview

The cats-vs-dogs model is a convolutional neural network (CNN) trained on the [Cats vs Dogs dataset](https://www.tensorflow.org/datasets/catalog/cats_vs_dogs),
a large set of images of cats and dogs.

### Details

- **Size:**  3,453,121 parameters
- **Model type:** CNN
- **Optimizer:** `RMSprop`
- **Number of Epochs:** 
- **Hardware:** Tesla V4
- **Emissions:** Not measured
- **Total Energy Consumption:** Not measured

### How to Use

```python
!pip install huggingface_hub["tensorflow"] -q

# We are loading our Keras model from the Hub via the `huggingface_hub.from_pretrained_keras`
from huggingface_hub import from_pretrained_keras
import tensorflow as tf

model = from_pretrained_keras("AiresPucrs/cats-vs-dogs")
model.trainable = False
model.summary()

```

## Intended Use

This model was created for research purposes only. We do not recommend any application of this model outside this scope.


## Performance Metrics

Accuracy

## Training Data

 [cats_vs_dogs dataset](https://www.tensorflow.org/datasets/catalog/cats_vs_dogs).

 The dataset consists of a set of images of cats and dogs.

 **Details:**

 - [Homepage](https://www.microsoft.com/en-us/download/details.aspx?id=54765)
 - Version: 1.0
 - Date Published: 5/9/2022
 - File Size:786.7 MB

## Training hyperparameters

The following hyperparameters were used during training:

| Hyperparameters | Value |
| :-- | :-- |
| name | RMSprop |
| learning_rate | 9.999999747378752e-05 |
| decay | 0.0 |
| rho | 0.8999999761581421 |
| momentum | 0.0 |
| epsilon | 1e-07 |
| centered | False |
| training_precision | float32 |
## Limitations
We do not recommend using this model in real-world applications. It was solely developed for academic and educational purposes.

## Cite as

```latex
@misc{teenytinycastle,
    doi = {10.5281/zenodo.7112065},
    url = {https://github.com/Nkluge-correa/teeny-tiny_castle},
    author = {Nicholas Kluge Corr{\^e}a},
    title = {Teeny-Tiny Castle},
    year = {2024},
    publisher = {GitHub},
    journal = {GitHub repository},
}
```

## License

The cats-vs-dogs pre-trained model is under the Apache 2.0 License.
The  cats_vs_dogs Dataset is licensed under the [Creative Commons(CC) License CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by/4.0/).

 ## Model Plot

<details>
<summary>View Model Plot</summary>

![Model Image](./model.png)

</details>