Research Article

Applying Bayesian Optimization for Machine Learning Models in Predicting the Surface Roughness in Single-Point Diamond Turning Polycarbonate

Table 4

Description of the optimization problems in this study.

ParameterFor XGBFor CAT

Decision variableslearning_ratelearning_rate
max_depthdepth
subsamplebagging_temperature
colsample_bytreenum_leaves
reg_alpha
max_leaves
gamma
min_child_weight
ObjectiveMinimize(test-RMSE)Minimize(test-RMSE)
Bounds of decision variables0.001 ≤ learning_rate ≤ 0.30.001 ≤ learning_rate ≤ 0.5
5 ≤ max_depth ≤ 305 ≤ depth ≤ 8
0.5 ≤ subsample ≤ 13 ≤ bagging_temperature ≤ 10
0.3 ≤ colsample_bytree ≤ 130 ≤ num_leaves ≤ 150
0.005 ≤ reg_alpha ≤ 0.02
0 ≤ max_leaves ≤ 0.02
0 ≤ gamma ≤ 1
1 ≤ min_child_weight ≤ 10