Skip to content

A machine learning model for predicting diabetes risk using key health indicators. The project involves data preprocessing, model development with TensorFlow and scikit-learn, and evaluation to achieve accurate predictions.

Notifications You must be signed in to change notification settings

ajitmane36/Diabetes-Prediction-Deep-Learning-Model

Repository files navigation

Diabetes-Prediction-Deep-Learning-Model

Overview

This repository contains a machine learning project focused on predicting diabetes risk. The project involves training various machine learning models, including Random Forest, SVM, and deep learning models. The best-performing model, Random Forest, achieved the highest F1 score and was identified as the optimal model for predicting diabetes risk.

Features

  • Data Preprocessing: Handling class imbalance, and splitting the dataset.
  • Model Training: Multiple ML models including Logistic Regression, Random Forest, SVM, KNN, Naive Bayes, and XGBoost.
  • Deep Learning: A custom-built Artificial Neural Network (ANN) model.
  • Model Evaluation: Metrics include F1 score, accuracy, confusion matrix, and classification report.
  • Model Compression: The best model is saved as a compressed file using the bz2file library to reduce its size for easy deployment and sharing.

Installation

  1. Clone the repository:
    git clone https://github.com/ajitmane36/Diabetes-Prediction-Deep-Learning-Model.git

About

A machine learning model for predicting diabetes risk using key health indicators. The project involves data preprocessing, model development with TensorFlow and scikit-learn, and evaluation to achieve accurate predictions.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published