The prediction and analysis of heart disease using XGBoost algorithm

Abstract

Heart diseases remain a global health concern, with their intricate aetiology and multifactorial risk factors making early diagnosis challenging. Recognizing the pressing need for accurate prediction tools, this research ventured into harnessing the power of machine learning, notably the Xtreme Gradient Boosting (XGBoost) algorithm, to fill this gap. The main object is to devise a robust predictive framework capable of early and accurate identification of heart disease. Specifically, our methodology unfolded systematically, beginning with data preprocessing, then delving into incisive feature selection, rigorous model training, and finally, thorough evaluation. This study is meticulously conducted on the heart.csv dataset, a comprehensive repository of cardiovascular data points. The experimental outcomes were nothing short of revelatory. Not only did the XGBoost model manifest superior performance metrics, but its precision also outpaced several contemporary models referenced in existing literature. Ultimately, our findings underscore the profound potential of the XGBoost algorithm in heart disease predictions. Beyond academic intrigue, this research holds tangible implications for healthcare practitioners, potentially offering a novel tool for early interventions and patient management.