Skip to main content
. 2021 Nov 11;57(11):1230. doi: 10.3390/medicina57111230

Table A1.

Optimal Hyperparameters of All Machine Learning Models.

Model Optimal Hyperparameters
LR nIter = 21
KNN k = 7
NB usekernal, Laplace = 0, Adjust = 1
DT Maximum depth = 5
Criterion = Gini index
RF Mtry * = 3
GBM Maximum depth = 3
Number of estimators = 50,
Gamma = 0
SVM degree = 3, scale = 0.1 and C = 1.0
ANN Number of hidden layers = 2
Number of nodes in a layer = 20, 10

LR, logistic regression; KNN, k-nearest neighbors; NB, naïve Bayes; DT, decision tree; RF, random forest; GBM, gradient boosting machine; SVM, support vector machine; ANN, artificial neural networks. * mtry indicates the number of variables available for splitting at each tree node.