Enhancing stroke prediction using the waikato environment for knowledge analysis

Abstract

State-of-the-art data mining tools incorporate advanced machine learning (ML) and artificial intelligence (AI) models, and it is widely used in classification, association rules, clustering, prediction, and sequential models. Data mining is important for the process of diagnosing and predicting diseases in the early stages, and this contributes greatly to the development of the health services sector. This study utilized classification to predict the stroke of a sample of the patient dataset that was taken from Kaggle. The classification model was created using the data mining program waikato environment for knowledge analysis (WEKA). This data mining tool helped identify individuals most at risk of stroke based on analysis of features extracted from the patient’s dataset. These features were used in classification processes according to the naive Bayes (NB), random forest (RF), support vector machine (SVM), and multi-layer perceptron (MLP) algorithms. Analysis of the classification results of the previous algorithms showed that the SVM outperformed other algorithms in terms of accuracy (94.4%), sensitivity (100%), and F-measure (97.1%). However, the NB algorithm had the best performance in terms of precision (95.7%).