执行网格搜索后如何分配模型参数?
现在,这就是我所做的:
model = ExtraTreesRegressor(n_estimators=10000, n_jobs=-1, random_state=0)
param_grid = {
'n_estimators': [1000, 2500, 5000, 7500],
'max_features': ['auto', 'sqrt', 'log2']
}
gs = GridSearchCV(estimator=model, param_grid=param_grid)
gs.fit(data, target)
logger.info(gs.best_params_)
model = gs.best_estimator_
我想将参数njobs=-1
分配给模型。网格搜索后如何执行此操作?
最佳答案
在您的代码中,生成的模型应该已将 n_jobs=-1
作为其参数之一,因为您在将模型传递给 GridSearchCV
之前已将其初始化为如此>。网格搜索将仅搜索您指定的参数,即 n_estimators
和 max_features
。
尽管如此,如果您想更新估算器的参数,您始终可以使用 set_params()
功能:
model.set_params(n_jobs=-1)
关于python - 执行网格搜索后分配模型参数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35169046/