Table 5 Comparison of prediction performance of three algorithm models

From: A color prediction model for mending materials of the Yuquan Iron Pagoda in China based on machine learning

Model

Training set

Validation set

Testing set

MAE

R2

MAE

R2

MAE

R2

ANN

1.56118

0.77323

1.06197

0.83368

0.74398

0.92604

XGBoost

0.06490

0.99967

1.11961

0.82886

0.68485

0.94238

LightGBM

1.08406

0.89241

1.45774

0.78006

0.90072

0.92315

Back to article page