时间戳

首页 > 解决方案 > 如何通过 GridSearchCV 打印最佳参数进行 k 折交叉验证

python - 如何通过 GridSearchCV 打印最佳参数进行 k 折交叉验证

问题描述

在使用 sklearn 预测具有 10 倍交叉验证的值时，我使用了 GridSearch 进行参数优化，如下所示，

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。我真的会得到一些帮助。提前致谢。

标签： pythonscikit-learncross-validationgrid-search

解决方案

最佳参数可用作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_)

推荐阅读