Predicting heart disease – a case study

You have been asked by a hospital to use machine learning to predict heart disease. Your job is to develop a model and highlight two to three important features that doctors and nurses can focus on to improve patient health.

You decide to use a decision tree classifier with fine-tuned hyperparameters. After the model has been built, you will interpret results using feature_importances_, an attribute that determines the most important features in predicting heart disease.

Heart Disease dataset

The Heart Disease dataset has been uploaded to GitHub as heart_disease.csv. This is a slight modification to the original Heart Disease dataset (https://archive.ics.uci.edu/ml/datasets/Heart+Disease) provided by the UCI Machine Learning Repository (https://archive.ics.uci.edu/ml/index.php) with null values cleaned up for your convenience.

Upload the file and display the first five rows as follows:

df_heart ...