ICU Mortality Prediction Model
An ICU mortality prediction project comparing multiple ML approaches on real patient data.
ICU Mortality Prediction Model
This project focused on predicting ICU patient mortality from a Kaggle healthcare dataset. I used Python, Pandas, and PyTorch to clean the data, compare multiple model families, and iterate toward the strongest-performing approach.
Highlights
- Built a full training pipeline for preprocessing, feature comparison, evaluation, and hyperparameter tuning
- Tested multiple architectures including logistic regression, random forest, and neural networks
- Chose the best-performing model based on precision and recall tradeoffs
- Reached approximately 99% accuracy during the final tuned runs
What I learned
Working with medical data made model evaluation more important than headline accuracy. Comparing multiple approaches and understanding where they performed well helped turn this into more than just a training exercise.
Predictive Model AUC Performance
ROC curve from one of the evaluated models, used to compare class separation performance during experimentation.