Analyzing the Impact of Data Resampling on Stroke Prediction using Machine Learning
This study focuses on stroke prediction using machine learning algorithms and evaluates the impact of different resampling techniques, including original, under-sampling, and over-sampling, on classification performance. The classifiers used in this study include Random Forest (RF), Decision Tree (DT), Gradient Boosting (GB), and K-Nearest Neighbor (KNN). Each model was trained and evaluated using performance metrics such as accuracy, precision, recall, F1-score, and AUC. The results demonstrate that RF trained on the oversampled dataset achieved the best performance with an accuracy of 94.31%, a precision of 93.52%, a recall of 95.27%, an F1-score of 94.39%, and an AUC of 98.46% on the test set. These findings highlight the effectiveness of oversampling in handling imbalanced datasets and the superiority of RF in stroke prediction tasks compared to other classification methods and resampling techniques.
Authors
- Majid Rahardi (ORCID: https://orcid.org/0000-0002-8692-302X)
- Afrig Aminuddin (ORCID: https://orcid.org/0000-0003-0963-5419)
- Ferian Fauzi Abdulloh (ORCID: https://orcid.org/0000-0003-3515-5240)
- Kusnawi Kusnawi (ORCID: https://orcid.org/0000-0002-2968-3744)
- Bima Pramudya Asaddulloh
- Hesmeralda Rojas Enriquez
Publication Details
- Journal
- Engineering Technology & Applied Science Research
- Published
- 2025-04-03
- DOI
- https://doi.org/10.48084/etasr.9736
- Citations
- 3
- Primary Topic
- Acute Ischemic Stroke Management
- Type
- article
- Field-Weighted Citation Impact
- 4.06