2016-12-25 8 views
0

私はSVSモデルに最適なパラメータ(C &ガンマ)を得るためにGridSearchCVを使用しています。結果、このコードで何が間違っていますか?GridSearchCV()によってScikitが最良のC&ガンマを見つけました__着いています

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_)) 

答えて

0

n_splits代わりに、それはsklearn.model_selection.ShuffleSplitためのparamであるsklearn.cross_validation.ShuffleSplitのPARAMではありません。 sklearn.cross_validationcode baseから

class ShuffleSplit(BaseShuffleSplit): 
    """Random permutation cross-validation iterator. 

    .. deprecated:: 0.18 
     This module will be removed in 0.20. 
     Use :class:`sklearn.model_selection.ShuffleSplit` instead. 
+0

私は更新され、エラーが固定されているが、私は今、別のエラーを持っている...例外TypeError「KFold」オブジェクトは、 –

+0

'cv'のparamは' KFold'を取るこ​​とができない反復可能ではありません入力として。 '5'を使いたい場合は' cv = 5'を入力し、 'KFold'や' StratifiedKFold'を使って分割を生成します。ドキュメント[here](http://scikit-learn.org/stable/modules/cross_validation.html#computing-cross-validated-metrics)を参照してください。 – tihom

関連する問題