【发布时间】:2020-02-11 10:30:19
【问题描述】:
我想使用 scikit-learn 的 StackedRegressor 来构建一个集成,但我想使用预训练的 keras 模型。基本上我想利用元回归器以最好的方式使用每个基本模型,但我不需要对它们进行训练。像
regressor_1 = KerasRegressor(build_fn)
regressor_2 = KerasRegressor(build_fn)
regressor_3 = KerasRegressor(build_fn)
regressors = [('r1', regressor_1),
('r2', regressor_2),
('r3', regressor_3)]
xgboost = XGBRegressor(learning_rate=0.01, n_estimators=3460,
max_depth=3, min_child_weight=0,
gamma=0, subsample=0.7,
colsample_bytree=0.7,
objective='reg:linear', nthread=-1,
scale_pos_weight=1, seed=27,
reg_alpha=0.00006)
stack_gen = StackingRegressor(estimators=regressors,
final_estimator=xgboost,
n_jobs=-1)
但同样没有拟合合适的基础模型。
【问题讨论】:
标签: python keras scikit-learn