Присвоение параметров модели после выполнения поиска по сетке

Question

Как я могу назначить параметры модели после выполнения поиска по сетке?

Прямо сейчас, это то, что я делаю:

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 к модели. Как мне сделать это после поиска по сетке?

Answer 1

В вашем коде полученная модель уже должна иметь n_jobs=-1 как один из его параметров, так как вы инициализировали модель так, прежде чем передать ее GridSearchCV, Поиск по сетке будет искать только те параметры, которые вы указали, а именно n_estimators а также max_features,

Тем не менее, если вы хотите обновить параметры оценщика, вы всегда можете использовать set_params() функция:

model.set_params(n_jobs=-1)