-
This is an end-to-end machine learning project that aims to predict whether or not a patient has heart disease based on his/her medical attributes.
-
Matplotlib was used for data visualization and Scikit-Learn's estimators for modeling.
-
RandomizedSearchCV
andGridSearchCV
were used to tune hyperparameters. -
Scikit-Learn's
LogisticRegression
estimator produced the highest cross-validated accuracy score of 83% compared toKNeighborsClassifier
andRandomForestClassifier
. -
The features with the most impact to the model are sex, exercise induced angina (exang), chest pain type (cp), thalassemia (thal), and number of major vessels (0-3) colored by flourosopy (ca).
-
Check out the project here.
-
Notifications
You must be signed in to change notification settings - Fork 0
srpineda/ml-project-heart-disease-classification
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
An end-to-end machine learning project that aims to predict whether or not a patient has heart disease based on his/her medical attributes.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published