Table 4 Average performance of the optimal models on the training and independent test datasets with different feature selection algorithms.
From: Machine learning–based method for analyzing stress distribution in a ship
Feature selection algorithm | Training dataset | Independent test dataset | ||||
|---|---|---|---|---|---|---|
RMSE | MAE | R² | RMSE | MAE | R² | |
XGBoost | 0.354 | 0.239 | 0.993 | 1.107 | 0.718 | 0.958 |
LightGBM | 0.407 | 0.262 | 0.993 | 5.993 | 2.092 | 0.921 |
CatBoost | 0.406 | 0.261 | 0.994 | 1.691 | 0.955 | 0.968 |
