Heart disease prediction using hyper parameter optimization (HPO) tuning

• Published in: Biomedical signal processing and control 2021-09, Vol.70, p.103033, Article 103033
• Main Authors: Valarmathi, R., Sheela, T.
• Format: Article
• Language: English
• Subjects: Grid search; Heart disease; Hyper parameter tuning; Randomized search; TPOT classifier
• ISSN: 1746-8094; 1746-8108
• DOI: 10.1016/j.bspc.2021.103033

•The Random forest classifier and XGBoost classifier model by Hyper Parameter Optimization.•The abnormal narrowing of heart vessel is angiography.•The combination of the ML models and optimization technique.•The machine learning models Random forest. Coronary artery disease prediction is considered to be one of the most challenging tasks in the health care industry. In our research, we propose a prediction system to detect the heart disease. Three Hyper Parameter Optimization (HPO) techniques Grid Search, Randomized Search and Genetic programming (TPOT Classifier) were proposed to optimize the performance of Random forest classifier and XG Boost classifier model. The performance of the two models Random Forest and XG Boost were compared with the existing studies. The performance of the models is evaluated with the publicly available datasets Cleveland Heart disease Dataset (CHD) and Z-Alizadeh Sani dataset. Random Forest along with TPOT Classifier achieved the highest accuracy of 97.52%for CHD Dataset. Random Forest with Randomized Search achieved the highest accuracy of 80.2%, 73.6% and 76.9% for the diagnosis of the stenos is of three vessels LAD, LCX and RCA respectively with Z-Alizadeh Sani Dataset. The results were compared with the existing studies focusing on prediction of heart disease that were found to outperform their results significantly.