Skip to main content

Table 3 Performance of Various Machine Learning Algorithms

From: Cluster energy prediction based on multiple strategy fusion whale optimization algorithm and light gradient boosting machine

Methods

MAE

MSE

RMSE

R2

Time (s)

RF

0.832

1.263

1.124

0.783

4.047

GBDT

0.751

1.059

1.029

0.841

1.941

XGBoost

0.728

0.986

0.993

0.849

1.425

LightGBM

0.714

1.024

1.012

0.855

0.445

MSFWOA-LightGBM

0.664

0.897

0.947

0.879

0.345