The Ensemble Learners: Machine Learning Classification of Diabetic Patient Readmission

Qianwen Ariel Xu, Victor Chang, Eniola Rahman Olamide, Meghana Ashok Ganatra, Wong Siu Tung

Research output: Chapter in Book/Published conference outputConference publication

Abstract

Diabetes is a major health problem that affects a large number of people worldwide. We proposed an effective and reliable method to diagnose diabetic readmission in this research based on machine learning models. The model was developed on tree-based ensemble classification algorithms, such as Decision Tree, Extreme Gradient Boost, Ada Boost, and CatBoost. In addition, to increase accuracy, we stacked the models with Stack Classifier using Catboost classifier as the final estimator. We carried out the experiments on the diabetic readmission dataset obtained from the VCI machine learning repository. We used the Grid Search technique to learn the best practices for model evaluation and hyperparameter tuning. The performances of all six algorithms are measured using different metrics such as AUC, Accuracy, and Recall. AUC measures a classifier's ability to distinguish between classes. CatBoost Classifier outperforms other algorithms in terms of AUC, Recall, and Accuracy, with 68 percent, 58.7 percent, and 63.2 percent, respectively. These findings are validated using ROC (Receiver Operating Characteristic) curves correctly and systematically.
Original languageEnglish
Title of host publicationProceedings - 2023 International Conference on Industrial IoT, Big Data and Supply Chain, IIoTBDSC 2023
PublisherIEEE
Pages76-85
Number of pages10
ISBN (Electronic)9798350341690
DOIs
Publication statusPublished - 18 Jun 2024

Publication series

NameProceedings - 2023 International Conference on Industrial IoT, Big Data and Supply Chain, IIoTBDSC 2023

Bibliographical note

Publisher Copyright:
© 2023 IEEE.

Keywords

  • AUC
  • Accuracy
  • CatBoost
  • Decision Tree
  • Diabetic Readmission
  • Ensemble
  • Machine Learning
  • Recall
  • Stack Classifier

Fingerprint

Dive into the research topics of 'The Ensemble Learners: Machine Learning Classification of Diabetic Patient Readmission'. Together they form a unique fingerprint.

Cite this