Abstract:Background: Early prediction and classification of prognosis is essential for patients in the coronary care unit (CCU). We applied a machine learning (ML) model using the eXtreme Gradient Boosting (XGBoost) algorithm to prognosticate CCU patients and compared XGBoost with traditional classification models. Methods: CCU patients' data were extracted from the MIMIC-III v1.4 clinical database, and divided into four groups based on the time to death: <30 days, 30 days−1 year, 1–5 years, and ≥5 years. Four classification models, including XGBoost, naïve Bayes (NB), logistic regression (LR), and support vector machine (SVM) were constructed using the Python software. These four models were tested and compared for accuracy, F1 score, Matthews correlation coefficient (MCC), and area under the curve (AUC) of the receiver operating characteristic curves. Subsequently, Local Interpretable Model-Agnostic Explanations method was performed to improve XGBoost model interpretability. We also constructed sub-models of each model based on the different categories of death time and compared the differences by decision curve analysis. The optimal model was further analyzed using a clinical impact curve. At last, feature ablation curves of the XGBoost model were conducted to obtain the simplified model. Results: Overall, 5360 CCU patients were included. Compared to NB, LR, and SVM, the XGBoost model showed better accuracy (0.663, 0.605, 0.632, and 0.622), micro-AUCs (0.873, 0.811, 0.841, and 0.818), and MCC (0.337, 0.317, 0.250, and 0.182). In subgroup analysis, the XGBoost model had a better predictive performance in acute myocardial infarction subgroup. The decision curve and clinical impact curve analyses verified the clinical utility of the XGBoost model for different categories of patients. Finally, we obtained a simplified model with thirty features. Conclusions: For CCU physicians, the ML technique by XGBoost is a potential predictive tool in patients with different conditions, and it may contribute to improvements in prognosis.

A Novel Ion Channel Prediction Method Based on XGBoost Model

Predicting Ion Channels Genes and Their Types with Machine Learning Techniques

A Novel Protein Subcellular Localization Method with CNN-XGBoost Model for Alzheimer's Disease.

Predicting the Prognosis of Patients in the Coronary Care Unit: A Novel Multi-Category Machine Learning Model Using XGBoost

Predict potential drug targets from the ion channel proteins based on SVM.

PSIONplus: Accurate Sequence-Based Predictor of Ion Channels and Their Types

Ionchanpred 2.0: A Tool To Predict Ion Channels And Their Types

Predicting the Prognosis of Patients in the Coronary Care Unit via Machine Learning Using XGBoost

Prediction of the Types of Ion Channel-Targeted Conotoxins Based on Feature Selection Techniques

Evaluating the predictive accuracy of ion channel models using data from multiple experimental designs

Prediction of the types of ion channel-targeted conotoxins based on radial basis function network.

Briefing in Application of Machine Learning Methods in Ion Channel Prediction

Identifying Ion Channel Genes Related to Cardiomyopathy Using A Novel Decision Forest Strategy

Prediction of Ion Channels and Their Types from Protein Sequences: Comprehensive Review and Comparative Assessment.

Improved Prediction Model of Protein and Peptide Toxicity by Integrating Channel Attention into a Convolutional Neural Network and Gated Recurrent Units

Predicting Potential Drug Targets from Ion Channel Proteins Based on Ensemble Learning

hERGBoost: A gradient boosting model for quantitative IC50 prediction of hERG channel blockers

Prediction of Blood-Brain Barrier Permeability of Compounds by Fusing Resampling Strategies and eXtreme Gradient Boosting

A Highly Efficient Biomolecular Network Representation Model for Predicting Drug-Disease Associations.

Predicting Ion Channels and Their Types by the Dipeptide Mode of Pseudo Amino Acid Composition.

Toxicity Prediction Method Based on Multi-Channel Convolutional Neural Network