scikit-learn - scikit-learn(sklearn) : TypeError: 'KFold' object is not iterable 中的 GridSearchCV

Question

我正在尝试使用 GridSearchCV() 为 SVR() 估计器找到 C 和 gamma 的最佳值，但出现此错误

TypeError: 'KFold' 对象不可迭代

这是代码

from sklearn.grid_search import GridSearchCV
from sklearn.model_selection import KFold
C_range = np.logspace(-2, 10, 13)
gamma_range = np.logspace(-9, 3, 13)
param_grid = dict(gamma=gamma_range, C=C_range)
cv = KFold(n_splits=5, shuffle=False, random_state=None)
grid = GridSearchCV(SVR(kernel='rbf'), param_grid=param_grid, cv=cv)
grid.fit(X, y)

print("The best parameters are %s with a score of %0.2f"
  % (grid.best_params_, grid.best_score_))

Answer 1

解决了类似的问题:

替换:

from sklearn.grid_search import GridSearchCV

与

from sklearn.model_selection import GridSearchCV