Prediction of stroke disease with demographic and behavioural data using random forest algorithm

SHOBAYO, Olamilekan, ZACHARIAH, Oluwafemi, ODUSAMI, Modupe Olufunke and OGUNLEYE, Bayode (2023). Prediction of stroke disease with demographic and behavioural data using random forest algorithm. Analytics, 2 (3), 604-617. [Article]

Documents
32238:620293
[thumbnail of analytics-02-00034.pdf]
Preview
PDF
analytics-02-00034.pdf - Published Version
Available under License Creative Commons Attribution.

Download (4MB) | Preview
Abstract
Stroke is a major cause of death worldwide, resulting from a blockage in the flow of blood to different parts of the brain. Many studies have proposed a stroke disease prediction model using medical features applied to deep learning (DL) algorithms to reduce its occurrence. However, these studies pay less attention to the predictors (both demographic and behavioural). Our study considers interpretability, robustness, and generalisation as key themes for deploying algorithms in the medical domain. Based on this background, we propose the use of random forest for stroke incidence prediction. Results from our experiment showed that random forest (RF) outperformed decision tree (DT) and logistic regression (LR) with a macro F1 score of 94%. Our findings indicated age and body mass index (BMI) as the most significant predictors of stroke disease incidence.
More Information
Statistics

Downloads

Downloads per month over past year

Metrics

Altmetric Badge

Dimensions Badge

Share
Add to AnyAdd to TwitterAdd to FacebookAdd to LinkedinAdd to PinterestAdd to Email

Actions (login required)

View Item View Item