Analisis Prediksi Stroke Menggunakan Pendekatan Decision Tree dengan Seleksi Fitur dan Neural Network

  • Indah Werdiningsih Universitas Airlangga
  • Endah Purwanti
  • Iin Mardiyana
  • Arum Tiyas Handayani
  • Kharristantie Sekarlangit Suryadewi
  • Endang Nurjanah
  • Fildzah Akhlaqulkarimah
  • Naurah Hedy Pramiyas
  • Fakhrana Almas Syah Yahrani
Keywords: Stroke, Machine Learning, Decision Tree, Neural Network

Abstract

Currently, stroke is the second cause of death globally. According to data from the World Health Organization (WHO), 7.9% of deaths in Indonesia are caused by stroke. Based on these data, analysis of the factors influencing the case growth rate is very useful. This paper analyzes various factors in electronic health records for effective stroke prediction with different machine-learning algorithms including Decision Tree and Neural Networks. This research uses a dataset consisting of 12 features, namely ID, gender, age, history of hypertension, history of heart disease, marital status, type of work, type of residence, average glucose level, BMI (Body Mass Index) number, and status. smoking, and prediction of stroke. These features were analyzed using the Neural Network and Decision Tree methods so that selected features were produced for further analysis using the Neural Network method. The feature selection results consist of 5 features: age, history of hypertension, marital status, average glucose level, and BMI (Body Mass Index) number. The highest accuracy results were obtained using the Neural Network method with a feature selection of 88.75, the second highest was obtained with the neural network method of 87.1875, and the lowest accuracy was obtained with the Decision Tree method which had an accuracy result of 81.25. Based on these accuracy results, it can be obtained that the most optimal results are shown by the Neural Network method with feature selection.

Downloads

Download data is not yet available.

References

Widyaswara suwaryo, P. A., Widodo, W. T., & Setianingsih, E. (2019). Faktor Risiko yang Mempengaruhi Kejadian Stroke. Jurnal Keperawatan (Vol. 11, No. 4, hal 251-260).

Mutiarasari, D., Kesehatan, B. I., Komunitas, M.- K., & Kedokteran, F. (2019). ISCHEMIC STROKE: SYMPTOMS, RISK FACTORS, AND PREVENTION. In Jurnal Ilmiah Kedokteran (Vol. 6, Issue 1).

Sailasya, G., & Kumari, G. L. A. (2021). Analyzing the Performance of Stroke Prediction using ML Classification Algorithms. International Journal of Advanced Computer Science and Applications (Vol. 12, Issue 6).

Budianita, E., Oktafalisa., & Ofandi, A. (2020). Penerapan Decision Tree dan Neural Network pada Penentuan Kategori Status Gizi Balita. Seminar Nasional Teknologi Informasi, Komunikasi dan Industri (SNTIKI) (Vol. 12).

Nurlia, Elin., & Enri, Ultach. (2021). Penerapan Fitur Seleksi Forward Selection Untuk Menentukan Kematian Akibat Gagal Jantung Menggunakan Algoritma C4.5. JUTIM (Jurnal Teknik Informatika Musirawas) (Vol. 6, No 1).

Nurlelah, E., Utami, Y, D. (2022). Seleksi Atribut Pada Algoritma Neural Network Menggunakan Particle Swarm Optimization Untuk Diagnosis Penyakit Liver. Kumpulan jurnaL Ilmu Komputer (KLIK) (Vol. 9, No. 2).

Kasih, Patmi. (2019). Pemodelan Data Mining Decision Tree Dengan Classification Error Untuk Seleksi Calon Anggota Tim Paduan Suara. Innovation In Research Of Informatics (Vol.1, No. 2, hal. 63-69).

Mohsena, A., Alsurorib, M., & Buthiena, A. (2019). New Approach to Medical Diagnosis Using Artificial Neural Network and Decision Tree Algorithm: Application to Dental Diseases. International Journal Information Engineering and Electronic Business (Vol 4).

S, Isha., & Borikar, A. (2022). Heart Disease Prediction using Data Mining Techniques. Journal Of Algebraic Statistics (Vol. 13, No.3).

Tigga, P., & Garg, S, (2020). Prediction of Type 2 Diabetes using Machine Learning Classification Methods. International Conference on Computational Intelligence and Data Science (ICCIDS 2019).

Pambudi, E. R., Sriyanto., & Firmansyah. (2022). Klasifikasi Penyakit Stroke Menggunakan Algoritma Decision Tree C.45. Jurnal Teknika (Vol. 16, No. 2, hal 221-226).

Drajana Rally, C. A., & Bode, A. (2022). Prediksi Status Penderita Stunting Pada Balita Provinsi Gorontalo Menggunakan K-Nearest Neighbor Berbasis Seleksi Fitur Chi Square. Jurnal Nasional Komputasi dan Teknologi Informasi (Vol. 5, No. 2).

Gupta, K., Jiwani, N., Afreen, N., & D, Divyarani. (2022). Liver Disease Prediction using Machine learning Classification Techniques. 11th IEEE International Conference on Communication Systems and Network Technologies.

Wahyuni, S., Elvira. (2016). Penerapan Metode Seleksi Fitur Untuk Meningkatkan Hasil Diagnosis Kanker Payudara. Jurnal SIMETRIS (Vol. 7, No. 1).

Prasetyo, Y., Simoen. (2023). Prediksi Gagal Jantung Menggunakan Artificial Neural Network. Jurnal SAINTEKOM (Vol.13, No. 1, hal. 79-88).

Rane, N., Sunny, J., Kanade, R., & Devi, S. (2020). Breast Cancer Classification and Prediction using Machine Learning. International Journal of Engineering Research & Technology (IJERT).

Nasser, M., Al-Shawwa., & Abu-Naser, S. (2019). A Proposed Artificial Neural Network for Predicting Movies Rates Category. International Journal of Academic Engineering Research (IJAER) (Vol. 3, Issue. 2).

Draper, N., Smith, H., 1992. Analisis Regresi Terapan Edisi Kedua, PT. Gramedia Pustaka Utama, Jakarta.

Fadly, M., Uddin, N., & Sutarto, H. Y., 2002. Flutter Suppression Using Neural Networks : Design and Implementation.

I. H. Witten, E. Frank, and M. a. Hall, 2011. Data Mining Practical Machine Learning Tools and Techniques Third Edition, vol. 277. Tentang Data Mining.

Stroke Prediction Dataset, Kaggle, May., 2022. [Online]. Available: https://www.kaggle.com/datasets/fedesoriano/stroke-prediction-dataset.

World Health Organization (WHO). 2019. 10 besar penyebab kematian. Diakses pada 25 April 2023, dari https://www.who.int/news-room/fact-sheets/detail/the-top-10-causes-of-death

World Health Organization (WHO). 2023. Stroke, Cerebrovascular accident. Diakses pada 25 April 2023, dari https://www.emro.who.int/health-topics/stroke-cerebrovascular-accident/index.html

Dev, S., Wang, H., Nwosu, C. S., Jain, N., Veeravalli, B., & John, D. (2022). A predictive analytics approach for stroke prediction using machine learning and neural networks. Healthcare Analytics, 2, 100032.

Rajora, M., Rathod, M., & Naik, N. S. (2021). Stroke prediction using machine learning in a distributed environment. In Distributed Computing and Internet Technology: 17th International Conference, ICDCIT 2021, Bhubaneswar, India, January 7–10, 2021, Proceedings 17 (pp. 238-252). Springer International Publishing.

Published
2023-12-06
How to Cite
Indah Werdiningsih, Purwanti, E., Mardiyana, I., Handayani, A. T., Suryadewi, K. S., Nurjanah, E., Akhlaqulkarimah, F., Pramiyas, N. H., & Yahrani, F. A. S. (2023). Analisis Prediksi Stroke Menggunakan Pendekatan Decision Tree dengan Seleksi Fitur dan Neural Network. Jurnal Sistem Cerdas, 6(3), 213 - 221. https://doi.org/10.37396/jsc.v6i3.310
Section
Articles