Skip to main content

Table 3 Cross-validation results

From: Development and validation of a machine learning model to predict time to renal replacement therapy in patients with chronic kidney disease

Algorithm

R2

(training fold)

MAE

(training fold)

R2 (cross-validation fold)

MAE (cross-validation fold)

Linear regression

0.72

418

0.50

531

Ridge regression

0.68

443

0.56

509

LASSO regression

0.69

427

0.59

488

Elastic net

0.68

439

0.58

500

Random forest

0.86

294

0.58

484

GBDT

0.93

198

0.62

459