Classification of Heart Disease Using Machine Learning Techniques


Rajendran, Perivitta and Haw, Su Cheng and Naveen, Palanichamy (2022) Classification of Heart Disease Using Machine Learning Techniques. In: ICDTE 2021: 2021 5th International Conference on Digital Technology in Education, 15 - 17 Sep. 2021, Busan, Republic of Korea.

[img] Text
Restricted to Repository staff only

Download (7MB)


The most crucial task in the medical field is diagnosing an illness. If a disease is determined at the early stage then many lives can be saved. The purpose of this paper is to use the medical data to predict cardiovascular heart disease using both supervised and unsupervised learning techniques and to show the effects of feature correlation on the classification model with over four different algorithms namely, Logistic Regression, Naive Bayes, Random Forest and Artificial Neural Networks. For the performance assessment, it incorporates F1-score, precision, Area under curve and recall. Overall, Logistic Regression algorithm tends to perform well for both Hungary and Statlog dataset whereas for Cleveland dataset, Artificial Neural Networks performs better than Logistic Regression in terms of accuracy. In terms of area under curve score, Logistic Regression performance is higher in all the dataset compared to Naive Bayes, Random Forest and Artificial Neural Networks. The results tabulated evidently prove that the designed diagnostic system is capable of predicting the risk level of heart disease effectively when compared to other approaches.

Item Type: Conference or Workshop Item (Paper)
Uncontrolled Keywords: Heart Disease, Logistic Regression, Naive Bayes, Random Forest, Artificial Neural Networks
Subjects: R Medicine > RC Internal medicine
Divisions: Faculty of Computing and Informatics (FCI)
Depositing User: Ms Nurul Iqtiani Ahmad
Date Deposited: 22 Feb 2022 03:41
Last Modified: 22 Feb 2022 03:41


Downloads per month over past year

View ItemEdit (login required)