---
language:
- en
library_name: tf-keras
license: apache-2.0
metrics:
- accuracy
pipeline_tag: image-classification
---
# Digit-classifier

## Model Overview

The digit-classifier model is used to recognize manuscript digits from 0 to 9.

### Details

- **Size:**  1,256,080  parameters
- **Model type:** Convolutional neural network
- **Optimizer**:  tf.keras.optimizers.Adam(learning_rate=0.001)
- **Number of Epochs:** 10
- **Batch size:** 256
- **Hardware:** Tesla V4
- **Emissions:** Not measured
- **Total Energy Consumption:** Not measured

### How to Use

```python
from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("AiresPucrs/digit-classifier")

classes = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
```

## Intended Use

This model was created for research purposes only. Specifically, it was designed to recognize manuscript digits from 0 to 9. We do not recommend any application of this model outside this scope.


## Training Data

The model was trained with the [MNIST digits classification dataset](https://keras.io/api/datasets/mnist/).


# 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


This model is licensed under the Apache License, Version 2.0.