Predicting the Chance of Heart Attack with a Machine Learning Approach – Supervised Learning

Abstract

Machine learning is a multidisciplinary field combining statistics, computer science and artificial intelligence. This research finds a way to use machine learning to predict the chance of heart attack based on information about the patient. There are 13 features collected about each patient which are age, sex, cholesterol, chest pain type, maximum heart rate achieved, resting blood pressure, resting electrocardiographic results, fasting blood sugar, exercise-induced angina, previous peak, slope, number of major blood vessels, and thalassemia. The information of all the patients is put into a dataset. The dataset is split into two sets, one for training and another for validation. A computer model using a supervised learning algorithm is developed and trained to predict the chance of heart attack. During training, the training set is used for training the model, while the validation set is used for evaluating the accuracy of the model.