This is a project for the Machine Learning course. This notebook includes classification of 918 patients with seven different machine learning algorithms.
- Dataset: Includes 11 features and 918 rows. Taken from Kaggle.
- Classification Methods: Naïve Bayes, SVM, Logistic Regression, Random Forest, XGBoost, LightGBM and MLP.
- heart.csv
- CS512_Final_Code.ipynb
- pandas
- numpy
- scikit-learn
- seaborn
- matplotlib
- xgboost
- lightgbm
Written with StackEdit.