Unveiling the potential of machine learning approaches in predicting the emergence of stroke at its onset: a predicting framework

Abstract

Abstract A stroke is a dangerous, life-threatening disease that mostly affects people over 65, but an unhealthy diet is also contributing to the development of strokes at younger ages. Strokes can be treated successfully if they are identified early enough, and suitable therapies are available. The purpose of this study is to develop a stroke prediction model that will improve stroke prediction effectiveness as well as accuracy. Predicting whether someone is suffering from a stroke or not can be accomplished with this proposed machine learning algorithm. In this research, various machine learning techniques are evaluated for predicting stroke on the healthcare stroke dataset. The feature selection algorithms used here are gradient boosting and random forest, and classifiers include the decision tree classifier, Support Vector Machine (SVM) classifier, logistic regression classifier, gradient boosting classifier, random forest classifier, K neighbors classifier, and Xtreme gradient boosting classifier. In this process, different machine-learning approaches are employed to test predictive methods on different data samples. As a result obtained from the different methods applied, and the comparison of different classification models, the random forest model offers an accuracy rate of 98%.

Keywords

• Random forest
• Logistic regression
• Gradient boosting classifier
• Decision tree
• Xtreme gradient boosting classifier