Как напечатать лучшие параметры через GridSearchCV для перекрестной проверки в k-кратном размере - PullRequest
0 голосов
/ 25 августа 2018

Я использовал GridSearch для оптимизации параметров при прогнозировании значений с 10-кратной перекрестной проверкой с использованием sklearn, как показано ниже,

svr_params = {
   'C': [0.1, 1, 10],
   'epsilon': [0.01, 0.05, 0.1, 0.5, 1],
    }

svr = SVR(kernel='linear', coef0=0.1, shrinking=True, tol=0.001, cache_size=200, verbose=False, max_iter=-1)
best_svr = GridSearchCV(
    svr, param_grid=svr_params, cv=10, verbose=0, n_jobs=-1)

predicted = cross_val_predict(best_svr, X, y, cv=10)

Я хочу распечатать лучшие параметры, выбранные GridSearch для C и epsilon.Я был бы действительно признателен за помощь.Заранее спасибо.

1 Ответ

0 голосов
/ 25 августа 2018

Лучшие параметры доступны как best_params_ атрибут GridSearchCV.

best_svr = GridSearchCV(svr, param_grid=svr_params, cv=10, verbose=0, n_jobs=-1, refit=True)
best_svr.fit(X, y)
print(best_svr.best_params_)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...