Skip to main content

Table 2 Tuning parameters of the predictive models

From: Comparison and development of machine learning tools in the prediction of chronic kidney disease progression

Models

Tuning

LR

α (Regularization parameter)

Elastic Net

γ (Mixing percentage),

α (Regularization parameter)

Lasso

α (Regularization parameter)

Ridge

α (Regularization parameter)

SVM

γ (Gaussian kernel), C (Cost)

RF

n_estimators (#subtrees)

k-NN

k (#Neighbors)

NN

Size (#hidden layer units),

α (Regularization parameter)

XGBoost

Depth (maximum depth of number), weight (the smallest sample weight and weight in the child node)