You can deploy this model and start earning money today!

mr-baraiya/
cardio-disease-model

Views

No views yet

Cardio Disease Prediction Model

A machine learning–powered Cardio Disease Prediction system using Random Forest and Logistic Regression, deployed with FastAPI and hosted on Hugging Face + Render.

This project predicts the risk of cardiovascular disease based on patient health parameters and provides both probability scores and risk classification.

Live API Demo

Base URL: https://cardio-fastapi-8ijy.onrender.com

Swagger UI: https://cardio-fastapi-8ijy.onrender.com/docs

Models Used

Random Forest Classifier
- Handles non-linear relationships
- Robust against noise
Logistic Regression
- Interpretable probabilities
- Baseline medical risk model

Both models are trained on a cleaned cardiovascular dataset.

Repository Contents


cardio-disease-model/
│
├── random_forest_model.pkl     # Trained Random Forest model
├── logistic_weights.npy        # Logistic regression weights
├── logistic_bias.npy           # Logistic regression bias
├── scaler_num.pkl              # Numerical feature scaler
├── scaler_int.pkl              # Categorical feature scaler
└── README.md

Input Features

Feature	Description
age	Age in years
gender	1 = Male, 0 = Female
height	Height (cm)
weight	Weight (kg)
ap_hi	Systolic blood pressure
ap_lo	Diastolic blood pressure
cholesterol	1=Normal, 2=Above Normal, 3=High
gluc	1=Normal, 2=Above Normal, 3=High
smoke	1=Smoker, 0=Non-smoker
alco	1=Alcohol intake, 0=No
active	1=Physically active, 0=Inactive

API Endpoints

Health Check


GET /health

Random Forest Prediction


POST /predict/randomforest

Logistic Regression Prediction


POST /predict/logistic

Compare Both Models


POST /predict/compare

Example Request

json

{
  "age": 55,
  "gender": 1,
  "height": 170,
  "weight": 72,
  "ap_hi": 130,
  "ap_lo": 85,
  "cholesterol": 1,
  "gluc": 1,
  "smoke": 0,
  "alco": 0,
  "active": 1
}

Example Response

json

{
  "model": "Random Forest",
  "prediction": 0,
  "probability": 0.15,
  "risk": "Low Risk"
}

Disclaimer

This project is for educational and research purposes only. It must not be used as a substitute for professional medical diagnosis.

Tech Stack

Python
Scikit-learn
NumPy
FastAPI
Uvicorn
Hugging Face Hub
Render Cloud

License

This project is licensed under the MIT License.

Author

Vishal Baraiya B.Tech CSE | Machine Learning & Full Stack Developer

⭐ If you find this project useful, please consider starring the repository!

mr-baraiya/cardio-disease-model