Неверный параметр rf для оценщика Pipeline - PullRequest
0 голосов
/ 16 февраля 2020

Я использую RandomForestRegressor с GridSearchCV. Я получаю эту ошибку:

   ValueError: Invalid parameter RandomForestRegressor for estimator Pipeline

Я не понимаю, почему. Это мой код:

 pipelines = {
        'rf': make_pipeline(StandardScaler(),
                            RandomForestRegressor(random_state=42)),


rf_hyperparameters = {
    'RandomForestRegressor__n_estimators': [100, 200, 300, 400, 500],
    'RandomForestRegressor__max_features': ['auto', 'sqrt', 0.33]
    }

hyperparameters = {
    'rf': rf_hyperparameters,
}

fitted_alternative_models = {}
for name, pipeline in pipelines.items():
    alt_model = GridSearchCV(pipeline, hyperparameters[name], cv=10, n_jobs=-1)
    alt_model.fit(X_train, y_train.values.ravel())

Спасибо!

1 Ответ

0 голосов
/ 16 февраля 2020

Я изменил:

rf_hyperparameters = {
    'RandomForestRegressor__n_estimators': [100, 200, 300, 400, 500],
    'RandomForestRegressor__max_features': ['auto', 'sqrt', 0.33]
    }

на:

rf_hyperparameters = {
     'randomforestregressor__n_estimators': [100, 200, 300, 400, 500],
     'randomforestregressor__max_features': ['auto', 'sqrt', 0.33]
}

и все работает.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...