TABLE 2.

Performance metrics for machine learning.

Model	MAE of test set	R 2 of trian test	MSE of train test	RMSE of train test
GBDT	2.3204	0.8653	0.7913	0.8896
XGBoost	2.4149	0.9123	0.5151	0.7177
DTR	2.4811	0.8582	0.8325	0.9124
ETR	2.0784	0.8254	1.0253	1.0126

GBDT, Gradient Boosting Decision Tree, XGBoost, eXtreme Gradient Boosting; DTR, Decision Tree Regression; ETR, Extra Tree Regression; MSE, Mean Square Error; MAE, Mean Absolute Error; RMSE, Root Mean Square Error.