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.

Illustrative mockup of the Disease Prediction System using Machine Learning interface.
More project photos (2)

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.

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

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 abstract (PDF)

Related guides

All guides
Technical illustration of a cloud server pushing event envelopes outward to web application endpoints for payments, dashboards and chat.Web development and IoT students who have built a REST API or an ESP32 project and now need external services (payments, GitHub, messaging) to notify their app when things happen

Webhooks Explained: Receive Data from Payments, GitHub and IoT

A webhook lets a service call your server the moment an event happens — payments, code pushes, form submissions. This guide explains the event anatomy, HMAC signature verification, the fast-acknowledge reliability pattern, local testing, and how webhooks compare with polling, WebSockets and SSE.

Read guide
Illustration of object tracking showing video frames with bounding boxes and persistent ID labels following people and vehicles, comparing motion prediction and appearance matching.B.E./B.Tech Computer Science and Electronics students building video analytics projects — people counting, vehicle tracking, sports analysis — who have detection working and need

Object Tracking: DeepSORT and ByteTrack Explained

Detection finds objects per frame; tracking keeps their identities across frames. This guide explains tracking-by-detection, Kalman motion models, DeepSORT's appearance embeddings vs ByteTrack's low-confidence box recovery, tracking metrics (HOTA, IDF1, ID switches), and the tuning parameters that determine real-world quality.

Read guide
Illustration of image segmentation showing U-Net's U-shaped encoder-decoder with skip connections producing pixel masks, alongside Mask R-CNN detecting instances with masks.B.E./B.Tech Computer Science and AI/ML students moving from image classification or detection to pixel-level understanding — medical imaging, defect detection, autonomous driving

Image Segmentation: U-Net and Mask R-CNN

When projects need pixel-level answers, segmentation delivers. This guide explains semantic vs instance vs panoptic segmentation, U-Net's encoder-decoder with skip connections, Mask R-CNN's parallel mask head, Dice and IoU evaluation, paired augmentation, and how to choose the right architecture for your data and question.

Read guide
Get a quotation