我正在处理回归量问题。我正在调整 XGBRegressor 模型的参数,所以我使用库 GPyOpt 来获得优化的参数。这些函数返回一个包含 5 个元素的数组和最小化的 MSE,即 1813。然后我尝试在模型中输入优化后的参数,然后模型的 MSE 返回 2810。我想知道为什么会这样?
我对库 GPyOpt 非常熟悉。关于我面临的这个问题的信息不多,所以我想知道这是因为我的粗心错误还是有什么我不明白的地方?
import GPyOpt
from GPyOpt.methods import BayesianOptimization
def cv_score(parameters):
parameters = parameters[0]
score = cross_val_score(
XGBRegressor(learning_rate=parameters[0],
gamma=int(parameters[1]),
max_depth=int(parameters[2]),
n_estimators=int(parameters[3]),
min_child_weight = parameters[4]),
x_train, y_train, scoring='neg_mean_squared_error').mean()
score = np.array(score)
return score
bds = [{'name': 'learning_rate', 'type': 'continuous', 'domain': (0, 1)},
{'name': 'gamma', 'type': 'continuous', 'domain': (0, 5)},
{'name': 'max_depth', 'type': 'discrete', 'domain': (1, 50)},
{'name': 'n_estimators', 'type': 'discrete', 'domain': (1, 300)},
{'name': 'min_child_weight', 'type': 'discrete', 'domain': (1, 10)}]
optimizer = BayesianOptimization(f=cv_score, domain=bds,
model_type='GP',
acquisition_type ='EI',
acquisition_jitter = 0.05,
exact_feval=True,
maximize=True)
optimizer.run_optimization(max_iter=20)
optimizer.x_opt
数组([ 0.56133897, 2.697656 , 50. , 300. , 10. ])
xgb_final_param = {'learning_rate': 0.56133897, 'gamma': 2.697656, 'max_depth': 50, 'n_estimators': 300, 'min_child_weight': 10}
xgb_final = SklearnExtra(clf = XGBRegressor(), seed = Seed, params = xgb_final_param)
xgb_final.fit(x_train, y_train)
evaluate(xgb_final, x_test, y_test) #evaluate returns MSE
我预计 MSE 大约在 1813 左右,但我得到了 2810。所以我想知道为什么