π« Excited to share my latest ML project β Heart Disease Predictor!
I built an end-to-end machine learning application that predicts the presence or absence of heart disease based on patient vitals like blood pressure, cholesterol, chest pain type, and more. The dataset is taken from a kaggle competition ( playground Season 6 episode 2) which i participated in. π Problem it solves: Early detection of heart disease risk is critical. This tool allows anyone to input patient data and instantly get a prediction along with a visual explanation of which factors contributed most to the result β making the model transparent and interpretable, not just a black box.
βοΈ Technologies Used: β’ Python β’ XGBoost β for high-accuracy classification β’ Optuna β for hyperparameter tuning (achieved ~95.5% AUC) β’ SHAP β for explainable AI / feature importance visualization β’ Scikit-learn β for preprocessing and evaluation β’ Joblib β for model serialization β’ Streamlit β for building and deploying the web app β’ GitHub β for version control and deployment
π Key highlights: β Tuned XGBoost model with Optuna over 25 trials β SHAP values to explain every single prediction β Clean, minimal UI deployable in one click β Fully deployed and accessible via a public URL
This project taught me how to take a model from a Jupyter notebook all the way to a live, interactive web application β a skill I believe every ML engineer needs.