---
license: apache-2.0
datasets:
- mnist
language:
- en
metrics:
- accuracy
library_name: keras
---
# Description
This model is a convolutional neural network built with transfer learning using the pre-trained model 'VGG16.' The 'block5_conv1' layer is retrained, and a final dense layer with 128 neurons is added. 

The model will be used as a preliminary step in solving Sudokus through linear programming. Model It is responsible for classifying the content of each sudoku cell:

- class_0: empty cell
- class_1: cell contains the number 1
- class_2: cell contains the number 2
- class_3: cell contains the number 3
- class_4: cell contains the number 4
- class_5: cell contains the number 5
- class_6: cell contains the number 6
- class_7: cell contains the number 7
- class_8: cell contains the number 8
- class_9: cell contains the number 9

The dataset is constructed with balanced classes using images from the famous "MNIST digits classification" dataset, as well as images of numbers written digitally.

# Dataset schema
The image size it is 28x28 pixels. After applying data augmentation to the dataset, the total number of images is as follows:
- Training images: 5,600
- Validation images: 2,400
- Test images: 2,000

Test Accuracy: 0.9810

# Other validations:
An initial validation is performed. It remains pending to increase the size of the validations to understand the reliability of the mode

<div style="text-align: center;">
<img src="https://i.imgur.com/kdj9udt.jpg" width="300">
</div>
</div>

The results of the inference are as follows:
<div style="text-align: center;">
<img src="https://i.imgur.com/U2MJzH6.jpg" width="500">
</div>