Cardiovascular disease prediction using recursive feature elimination and gradient boosting classification techniques

Abstract

Abstract Cardiovascular diseases are one of the most common chronic illnesses that affect people's health. Early detection of cardiovascular diseases's can reduce mortality rates by preventing or reducing the severity of the disease. Machine learning algorithms are a promising method for identifying risk factors. This article proposes a recursive feature elimination‐based gradient boosting algorithm in order to obtain accurate heart disease prediction. The patients' health record with important cardiovascular disease features has been analysed for the evaluation of the results. Several other machine learning methods were also used to build the prediction model, and the results were compared with the proposed model. The results of this proposed model infer that the combined recursive feature elimination and gradient boosting algorithm achieves the highest accuracy (89.7%). Further, with an area under the curve of 0.84, the proposed algorithm was found superior and had obtained a substantial gain over other techniques. Thus, the proposed gradient boosting algorithm will serve as a prominent cardiovascular disease estimation and treatment model.