Abstract
To explore the use of different machine learning models in prediction of COVID-19 mortality in hospitalized patients. A total of 44,112 patients from six academic hospitals who were admitted for COVID-19 between March 2020 and August 2021 were included in this study. Variables were obtained from their electronic medical records. Random forest-recursive feature elimination was used to select key features. Decision tree, random forest, LightGBM, and XGBoost model were developed. Sensitivity, specificity, accuracy, F-1 score, and receiver operating characteristic(ROC)-AUC were used to compare the prediction performance of different models. Random forest-recursive feature eliminationselected following features to include in the prediction model: Age, sex, hypertension, malignancy, pneumonia, cardiac problem, cough, dyspnea, and respiratory system disease. XGBoost and LightGBM showed the best performance with an ROC-AUC of 0.83 [0.822-0.842] and 0.83 [0.816-0.837] and sensitivity of 0.77. XGBoost, LightGBM, and random forest have a relatively high predictive performance in prediction of mortality in COVID-19 patients and can be applied in hospital settings, however, future research are needed to externally confirm the validation of these models.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.