Research Article

Optimization of Tree-Based Machine Learning Models to Predict the Length of Hospital Stay Using Genetic Algorithm

Table 1

Details of reviewed studies.

Ref.SampleLOS typePredictive methodHPOBest modelValidation indicator
ContinuousCategoricalRegressionMachine learningDeep learning

[13]All admitted patientsMRMRR 20.267
[14]ICU-following cardiac surgery patientsLRLRAccuracy0.79
[10]Diabetic patientsGLM-NBRF-GBMANNVarious experimentsEnsembleAUC0.81
[17]NewbornsNB- LR- MLPDT-RF-R and.T-SVMNot mentionedRFAUC0.87
[4]Cardiac patientsRF-SVM-BNANNNot mentionedRFAccuracy0.8
[12]Neurosurgery patientsRNN-GRURNN-GRUMAE2.8
[15]Radical cystectomy for muscle-invasive bladder cancerGLMGLMR 20.048
[11]Brain tumor surgery patientsRegression-based modelsTree-based modelsANNGrid searchEnsembleRMSE0.5
[16]Elective patientsNB-BN -KNN- Kstar- LWLC4.5 DT-SVM-decision tableNot mentionedBNAUC0.9
[18]ICUFL -KNN-Rerg.T-NBCT-TB-RF-SVMANNNot mentionedFLAccuracy0.92
[20]COVID-19 patientsDTNot mentionedDTMAE2.84
[21]InpatientsNB- KNNDT -SVM,-XGBoost- nonlinear weighted XGBoostGrid searchNonlinear weighted XGBoostRMSE1.52
[13]Lung cancer patientsLRRF- XGBoostNot performedRF

Methods. BL, binary logistic; BN, Bayesian network; FL, fuzzy logic; GLM, generalized linear model; KNN, K-nearest neighbors; LR, logistic regression; LWL, locally weighted learning; MLP, multilayer perceptron; MR, multiple regression; NB, Naïve Bayes; Reg.T, regression tree; CT, classification trees; DT, decision tree (J48); GBM, gradient boosting machine; Rand.T, random tree; RF, random forest; SVM, support vector machines; TB, tree bagger; ANN, artificial neural network; GRU, gated recurrent unit; RNN, recurrent neuronal network.