A 2-Tier Stacking Ensemble Classifier for Disease Classification
- 1 Department of Computational Intelligence, Faculty of Engineering and Technology, SRM Institute of Science and Technology, Kattankulathur, Chennai, India
Abstract
Diarrhea, dysentery, and dehydration are top of the list of high mortality-causing diseases in children under the age group of five. Despite the tremendous growth of machine learning in the field of medical research, still, some areas remain untouched, especially, the pediatric department. Since medical data is used in the proposed work, even a slight increase in the accuracy of the model will be of great importance. In this study, we propose a 2-tier stacking ensemble method for disease identification. Initially, the data is pre-processed and sent to train the tier-1 machine learning models. Based on the majority voting the metadata is selected and sent to the next tier and finally, at the meta-classifier level disease classification happens. Performance metrics like the accuracy, precision, recall, F1-score and mean absolute error of the individual machine learning algorithms were analyzed and were used to compare with the proposed stacking ensemble method. The results proved that the 2-tier stacking model proposed in the work shows an accuracy of 95.41%, a precision of 94.47%, a recall value of 92.78%, and an F1-score of 93.50%. The proposed model achieved a high accuracy value when compared to the other machine learning models.
DOI: https://doi.org/10.3844/jcssp.2023.796.811
Copyright: © 2023 K. Sukanya Varshini and R. Annie Uthra. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
- 1,828 Views
- 1,062 Downloads
- 0 Citations
Download
Keywords
- Machine Learning
- Disease Identification
- Ensemble Stacking Model
- Dysentery
- Diarrhea
- Dehydration
- Pediatric