Stroke Prediction with Enhanced Gradient Boosting Classifier and Strategic Hyperparameter

Abstract

A stroke is a medical condition that occurs when the blood supply to the brain is interrupted. Stroke can cause damage to the brain that can potentially affect a person's function and ability to move, speak, think, and feel normally. The effect of stroke on health emphasizes the importance of stroke detection, so an effective model is needed in predicting stroke. This research aimed to find a new approach that can improve the performance of stroke prediction by comparing four derivative algorithms from Gradient Boosting by adding hyperparameters tuning. The addition of hyperparameters was used to find the best combination of parameter values that can improve the model accuracy. The methods used in this research were Categorical Boosting, Histogram Gradient Boosting, Light Gradient Boosting, and Extreme Gradient Boosting. The research involved retrieving, cleaning, and analyzing data and then the model performance was evaluated with a confusion matrix and execution time. The results obtained were Light Gradient Boosting with Hyperparameter RandomSearchCV achieved the highest accuracy at 95% among the algorithms tested, while also being the fastest in execution. The contribution of this research to the medical field can help doctors and patients predict the occurrence of stroke early and reduce serious consequences.