Application of the Naive Bayes Algorithm in Data Mining for Predicting Stroke Disease
DOI:
https://doi.org/10.25047/ijhitech.v4i1.6669Keywords:
Naive bayes, Data mining, StrokeAbstract
Stroke is a major global health problem that may lead to permanent disability or death. In Indonesia, analytical approaches such as data mining have been increasingly used to support early identification of stroke risk based on patient health records. This study applies the Naive Bayes algorithm as the main classification method. The research procedure includes data collection from the Kaggle repository, data selection based on predetermined clinical criteria, data cleaning to remove duplicates and missing values, and data transformation by converting categorical attributes into numerical form. The dataset was then split into training and testing subsets for model development. The final dataset consisted of 3,256 patient records containing variables such as gender, age group, hypertension, heart disease, average glucose level, body mass index (BMI), smoking status, and stroke occurrence. After completing these preprocessing stages, the Naive Bayes model achieved an accuracy of 89.49%. This result indicates that the model was able to classify stroke and non-stroke cases in the dataset with a satisfactory level of accuracy. The findings suggest that the model may serve as a baseline component for developing decision-support systems aimed at facilitating early identification of stroke risk in targeted population groups.
References
[1] A. S. Arifianto, M. Sarosa, and O. Setyawati, “Klasifikasi Stroke Berdasarkan Kelainan Patologis dengan Learning Vector Quantization,” J. EECCIS Electr. Electron. Commun. Controls Inform. Syst., vol. 8, no. 2, pp. 117–122, Aug. 2014, doi: 10.21776/jeeccis.v8i2.248.
[2] E. S. Donkor, “Stroke in the 2 1 s t Century: A Snapshot of the Burden, Epidemiology, and Quality of Life,” Stroke Res. Treat., vol. 2018, pp. 1–10, Nov. 2018, doi: 10.1155/2018/3238165.
[3] V. L. Feigin et al., “World Stroke Organization: Global Stroke Fact Sheet 2025,” Int. J. Stroke, vol. 20, no. 2, pp. 132–144, Feb. 2025, doi: 10.1177/17474930241308142.
[4] O. Ozaltin, O. Coskun, O. Yeniay, and A. Subasi, “A Deep Learning Approach for Detecting Stroke from Brain CT Images Using OzNet,” Bioengineering, vol. 9, no. 12, p. 783, Dec. 2022, doi: 10.3390/bioengineering9120783.
[5] L. Rahmawati, M. Hafid, and M. A. Sunandar, “Analisis Data Mining Untuk Memprediksi Penyakit Stroke Dengan Algoritma Naïve Bayes,” vol. 6, no. 2, 2023, doi: 10.17509/jatikom.v6i2.49051.
[6] M. Langarizadeh and F. Moghbeli, “Applying Naive Bayesian Networks to Disease Prediction: a Systematic Review,” Acta Inform. Medica, vol. 24, no. 5, p. 364, 2016, doi: 10.5455/aim.2016.24.364-369.
[7] P. Chakraborty et al., “Predicting stroke occurrences: a stacked machine learning approach with feature selection and data preprocessing,” BMC Bioinformatics, vol. 25, no. 1, p. 329, Oct. 2024, doi: 10.1186/s12859-024-05866-8.
[8] A. Z. Arrayyan, H. Setiawan, and K. T. Putra, “Naive Bayes for Diabetes Prediction: Developing a Classification Model for Risk Identification in Specific Populations,” Semesta Tek., vol. 27, no. 1, pp. 28–36, Apr. 2024, doi: 10.18196/st.v27i1.21008.
[9] M. Hassan et al., “Machine Learning-Based Rainfall Prediction: Unveiling Insights and Forecasting for Improved Preparedness”.
[10] M. Abbas, K. A. Memon, A. A. Jamali, S. Memon, and A. Ahmed, “Multinomial Naive Bayes Classification Model for Sentiment Analysis”.
[11] P. Zhang et al., “Design of an Automatic Classification System for Educational Reform Documents Based on Naive Bayes Algorithm,” Mathematics, vol. 12, no. 8, p. 1127, Apr. 2024, doi: 10.3390/math12081127.
[12] Kementerian Kesehatan Republik Indonesia, “Keputusan Menteri Kesehatan Republik Indonesia Nomor HK.01.07/MENKES/33/2025 tentang Petunjuk Teknis Pemeriksaan Kesehatan Gratis Hari Ulang Tahun,” HK.01.07/MENKES/33/2025, 2025. Accessed: Nov. 26, 2025. [Online]. Available: https://peraturan.bpk.go.id/Download/393217/Permendagri%20No.%2014%20Tahun%202025.pdf
[13] American Diabetes Association Professional Practice Committee et al., “2. Diagnosis and Classification of Diabetes: Standards of Care in Diabetes—2024,” Diabetes Care, vol. 47, no. Supplement_1, pp. S20–S42, Jan. 2024, doi: 10.2337/dc24-S002.
[14] World Health Organization, “WHO Discussion Paper: Obesity – Key Analysis,” Discussion Paper, 2021. Accessed: Nov. 26, 2025. [Online]. Available: https://cdn.who.int/media/docs/default-source/obesity/who-discussion-paper-on-obesity---final190821.pdf
[15] G. Sailasya and G. L. A. Kumari, “Analyzing the Performance of Stroke Prediction using ML Classification Algorithms,” Int. J. Adv. Comput. Sci. Appl., vol. 12, no. 6, 2021, doi: 10.14569/IJACSA.2021.0120662.
[16] A. G. Ramadhan, H. S. Hidayatullah, and R. Ruba’i, “Implementasi Algoritma C4.5 pada Sistem Prediksi Stroke Berdasarkan Data Kesehatan,” 2025.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2026 Nur Aggun S., Nadhifa Dwi Rahmalia, Nurrahma Harris, Wita Windari, Andi Muhammad Zulkifli, Mochammad Choirur Roziqin, Ziani Said
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
