Skip to content

This repository implements Multinomial Logistic Regression model for digit classification using the MNIST dataset. Built with TensorFlow, the workflow includes data loading, model training, and evaluation. The model is evaluated based on accuracy metrics, demonstrating the application of multinomial regression for image classification tasks.

Notifications You must be signed in to change notification settings

alkkuma1/MultinomialRegressionModelforClassification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

Multinomial Regression Model for Classification

Project Description

This repository contains a Python implementation of a Multinomial Logistic Regression model, which is used for classification tasks. The model is built and trained using the TensorFlow library on the MNIST dataset, a popular dataset for digit classification.

Table of Contents

  1. Installation
  2. Usage
  3. Dataset
  4. Model
  5. Results
  6. License

Installation

To run this project, you will need to have Python installed along with the following libraries:

  • pandas
  • tensorflow
  • numpy
  • matplotlib

You can install the necessary dependencies using the following command:

pip install pandas tensorflow numpy matplotlib

Usage

  1. Clone the repository:
    git clone https://github.com/yourusername/MultinomialRegressionModelforClassification.git
  2. Navigate to the project directory:
cd MultinomialRegressionModelforClassification
  1. Open and run the Jupyter Notebook:
jupyter notebook MultinomialRegressionModelforClassification.ipynb

The notebook walks through the entire process of building, training, and evaluating a Multinomial Regression model for digit classification using the MNIST dataset.

Dataset

The dataset used in this project is the MNIST dataset, which is loaded directly from the TensorFlow/Keras library. The dataset contains 70,000 grayscale images of handwritten digits (0-9), split into 60,000 training images and 10,000 test images.

MNIST = tf.keras.datasets.mnist

Model

The model implemented is a Multinomial Logistic Regression model, which is trained on the MNIST dataset to predict the digit represented in an image. The main steps include:

Loading and preparing the dataset Defining the model using TensorFlow Training the model using cross-entropy loss Evaluating the model's accuracy on test data

Results

After training the model, its performance is evaluated based on accuracy metrics, and the results are visualized using confusion matrices and plots of model performance.

License

This repository is public.

About

This repository implements Multinomial Logistic Regression model for digit classification using the MNIST dataset. Built with TensorFlow, the workflow includes data loading, model training, and evaluation. The model is evaluated based on accuracy metrics, demonstrating the application of multinomial regression for image classification tasks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published