Performance Evaluation of Supervised Machine Learning Algorithms in Prediction of Heart Disease

Abstract

Big challenge in health care industry is to record and analyze the massive amount of information about patients. Innovations in technologies made revolution in the healthcare industries. In recent years the data analytics developed as promising tool for problem solving and decision making in healthcare professions. Data analytics process the data automatically to make healthcare system more dynamic and robust. It systematically uses and analyses the data of health care for better treatment with low costs. The chief applications of Machine learning in healthcare are the detection and diagnosis of diseases. The heart is the chief organ of human body. Heart disease increases the mortality rate in the world. Around 90% of heart diseases are preventable. Machine learning plays a remarkable role in the health care industry in prediction of heart disease. In this research paper, the presence of heart disease is predicted by employing Decision Tree, Naive Bayes, Random Forest, Support Vector Machine, K-Nearest Neighbor and logistic Regression algorithms. The performance of the algorithms was analyzed using parameters such as Accuracy, Precision, AUC and F1-score. From the experimental result, it is found that the Random Forest is more accurate for predicting the heart disease with accuracy of 83.52% compared with other supervised machine learning algorithms. The F1- Score, AUC and precision score of Random forest classifiers are 84.21%, 88.24% and 88.89% respectively.