Heart Disease Prediction

This is a project for the Machine Learning course. This notebook includes classification of 918 patients with seven different machine learning algorithms.

Dataset and Machine Learning Models ❤️‍🩹

• Dataset: Includes 11 features and 918 rows. Taken from Kaggle.
• Classification Methods: Naïve Bayes, SVM, Logistic Regression, Random Forest, XGBoost, LightGBM and MLP.

Files

• heart.csv
• CS512_Final_Code.ipynb

Libraries

1. pandas
2. numpy
3. scikit-learn
4. seaborn
5. matplotlib
6. xgboost
7. lightgbm

Written with StackEdit.