File size: 967 Bytes
a3b9676 662d392 a3b9676 662d392 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
license: apache-2.0
datasets:
- mnist
metrics:
- accuracy
pipeline_tag: image-classification
---
# CNN Model for MNIST Digit Classification
This repository contains a Convolutional Neural Network (CNN) model trained on the MNIST dataset for digit classification. The model has achieved an accuracy of 99% on the test dataset and is available for use as a TensorFlow model.
## Model Details
- **Architecture**: Convolutional Neural Network (CNN)
- **Input Size**: 28x28 pixels with 1 channel (grayscale)
- **Data Preprocessing**: The model has been trained on normalized data.
- **Accuracy**: 99%
## Usage
You can use this model for digit classification tasks. Below are some code snippets to help you get started:
```python
# Load the model and perform inference
import tensorflow as tf
model = tf.keras.models.load_model('model.h5')
# Perform inference
predictions = model.predict(image)
# Get the predicted digit
predicted_digit = np.argmax(predictions)
|