mithun-ai/strokeprediction_XGBoost

Stroke Risk Predictor - XGBoost

Overview

This model predicts stroke risk using healthcare-related patient features.

Model Type

XGBoost Classifier

Inputs

• Age
• Hypertension
• Heart Disease
• Average Glucose Level
• BMI
• Smoking Status
• Other health indicators

Output

• 0 → No Stroke Risk
• 1 → Stroke Risk

Performance

Accuracy: 91.39%

Recall: 26%

ROC-AUC: 78.20%

Usage

import joblib

model = joblib.load("xgboost_model.joblib")

prediction = model.predict(data)

Author

Mithun