Detection of Cardiovascular Disease Using Ensemble Feature Engineering With Decision Tree

Abstract

Cardiovascular diseases are a cluster of heart-related issues, including many comorbidities, which are becoming a leading cause of human death across the globe. Hence, an essential framework is demanded for the early detection of CVDs which can help to prevent premature death. The application of Artificial Intelligence (AI) in healthcare has opted for this challenge and makes it easier to detect CVDs using a computational model. In this study, the authors built a reduced dataset using ensemble feature selection methods and got five features as per their weight values. Support Vector Machine, Logistic Regression, and Decision Tree classification techniques are utilized to check the effectiveness of newly designed datasets through different validation approaches. The authors also worked on data processing and visualization techniques, including Principal Component Analysis (PCA), and T-sne for understanding the data structure. From the findings, it was possible to conclude that DT has achieved an optimal accuracy and AUC of 98.9% and 0.99 ROC with leave one out Cross Validation (CV).