GRENZE International Journal of Engineering and Technology
Vol. 10
(2024), Issue 1
Revolutionizing Cerebral Stroke Prediction: Mastery Unveiled Through Stratified K-Fold and K-Fold Cross Validation Techniques for Imbalanced Datasets
Authors
Zarinabegam Mundargi, Sanskruti Khedkar, Sanket Kumbhar, Khushi Mohod, Yashashri Meshram
Abstract
In the realm of healthcare data analysis, stroke prediction stands as a critical task necessitating accurate and reliable machine learning models. Cross-validation techniques play a pivotal role in ensuring the robustness of these models. Among various methods, K-Fold and Stratified K-Fold have emerged as key contenders. The challenge of handling imbalanced datasets is particularly pertinent in stroke prediction, where the occurrence of strokes is often a rare event. This study meticulously explores the significance of cross-validation techniques in stroke prediction, with a focus on addressing class imbalances. Our research focuses on comparing the effectiveness of K-Fold and Stratified K-Fold techniques in managing the stroke dataset. Our findings shed light on the superiority of Stratified K-Fold in effectively managing imbalanced data, providing crucial insights for the advancement of stroke prediction models and healthcare applications
Pages:
2407 - 2413