metadata

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:

# 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)