Built to order

Disease Prediction System using Machine Learning

A machine-learning system that predicts diabetes, heart-disease and stroke risk from clinical input features. Trains and compares multiple classifiers with cross-validation, explains predictions with feature importance, and wraps it in a clean symptom-checker web app patients and examiners can try live.

Computer / ITAI & Machine LearningWeb Development PythonCloud DatabaseWeb DashboardMachine LearningRandom ForestData Sciencescikit-learnHealthcare AISoftware
Illustrative mockup of the Disease Prediction System using Machine Learning interface.
More project photos (3)

The problem

Disease Prediction System using Machine Learning — MediPredict is a complete ML-based healthcare prediction project that estimates the risk of diabetes, heart disease and stroke from clinical input features. It trains and compares multiple classifiers with cross-validation, explains every prediction with feature importance and SHAP values, and packages everything in the MediPredict symptom-checker web app that patients and examiners can try live during the demo. This is an ideal final year project for CSE, B.Tech and M.Tech students: it combines a real-world healthcare problem, a rigorous model-comparison methodology, explainable AI, and a polished working application — everything an examiner looks for.

SYSTEM FLOW

4 components 4 connections

Live flow
Drag to explore
100%
userPatient

Enters clinical values such as vitals and lab readings into the MediPredict app.

Follow the working sequence

How it works

  1. Data loading — standard clinical datasets for diabetes, heart disease (UCI) and stroke prediction are loaded and documented.
  2. Preprocessing pipeline — missing-value imputation, categorical encoding, feature scaling and a stratified train–test split keep the experiment reproducible.
  3. Model training & comparison — five classifiers are trained and evaluated with stratified k-fold cross-validation; the best model per disease is selected on F1-score and ROC-AUC.
  4. Model packaging — winning models are serialized and loaded by the MediPredict app — no GPU needed.
  5. Live prediction — user inputs are run through the same preprocessing pipeline; the app returns a risk probability, a severity band and an explanation of the top contributing features.
  6. Evaluation view — examiners can open the dashboard to inspect cross-validation metrics, confusion matrices and ROC curves for every model.

Project features

Multi-disease risk prediction — predicts diabetes, heart-disease and stroke risk from clinical input features in one system.
Live symptom-checker web app (MediPredict) — enter vitals and clinical values, get an instant risk probability with a severity band (Low / Moderate / High).
Model comparison built in — trains Logistic Regression, SVM, Random Forest, Gradient Boosting and k-NN, and compares them with stratified k-fold cross-validation.
Explainable predictions — every prediction shows per-feature importance and SHAP values, so the examiner can see why the model flagged the risk.
Evaluator dashboard — accuracy, precision, recall, F1-score, ROC-AUC, confusion matrices and ROC curves, all plotted and ready for the report.
Batch prediction — upload a CSV of patient records and screen them all at once.
Pre-trained models included — serialized .pkl model files bundled, so the demo runs instantly with no retraining.
Medical disclaimer on every prediction screen — responsible, examiner-friendly design.

What is included

Full source code (training pipeline + MediPredict web app)
Trained model files (.pkl) — demo-ready out of the box
Datasets, preprocessing scripts and experiment notebooks
Formatted project report PDF with metrics tables and plots
Viva presentation (PPT)
Viva Q&A preparation document
Demo walkthrough video
Installation & run guide
Viva support until your submission is complete

Advantages

  • Multi-disease risk prediction — diabetes, heart-disease and stroke risk from clinical input features in one system.
  • Trains and compares five classifiers (Logistic Regression, SVM, Random Forest, Gradient Boosting, k-NN) with stratified k-fold cross-validation, so model choice is justified transparently.
  • Explainable predictions — every prediction shows per-feature importance and SHAP values, so the examiner can see why a risk was flagged.
  • Evaluator dashboard with accuracy, precision, recall, F1-score, ROC-AUC, confusion matrices and ROC curves, all plotted and ready for the report.
  • Batch prediction — upload a CSV of patient records and screen them all at once.
  • Pre-trained .pkl models are bundled, so the demo runs instantly with no retraining; every prediction screen carries a medical disclaimer.

Disadvantages

  • MediPredict is an educational screening aid, not a medical diagnostic tool — predictions are probabilistic and never a substitute for professional medical advice.
  • Models are trained on public clinical datasets that may not represent all populations; generalization limits are documented in the report.
  • Accuracy is bounded by dataset quality and class imbalance — handled and reported transparently, not hidden.
  • The web app is a demonstration build for academic evaluation, not a certified clinical device.

Limitations & prerequisites

MediPredict is an educational screening aid, not a medical diagnostic tool — predictions are probabilistic and never a substitute for professional medical advice; the app carries a disclaimer on every prediction screen.
Models are trained on public clinical datasets that may not represent all populations; generalization limits are documented in the report.
Accuracy is bounded by dataset quality and class imbalance — handled and reported transparently, not hidden.
The web app is a demonstration build for academic evaluation, not a certified clinical device.

Components & software requirements

Python, scikit-learn, pandas, NumPy, SHAP / feature-importance explainability, Flask/Streamlit web app, Matplotlib/Seaborn (plots), Jupyter notebooks (experiments), pickle/joblib (model serialization)

Download free synopsis (PDF)

Enter your name and WhatsApp number — the synopsis PDF downloads immediately. We may send occasional project updates on WhatsApp; no spam.

Mohan, Thirumalai & Srivastava (2019), IEEE Access, vol. 7

Get a quotation