Skip to content

Part 2 - Machine Learning Models Integration

Latest
Compare
Choose a tag to compare
@sehroz sehroz released this 06 Dec 04:31
· 12 commits to main since this release
8a46447

This release introduces the integration of machine learning models for the Heart Failure Prediction project.

ML Models Integrated:

XGBoost (best-performing model with 92% accuracy).
Logistic Regression and Neural Networks are also implemented for comparative analysis.

Code Updates:

Model training, testing, and evaluation scripts included.
Feature importance analysis added to help explain predictions.