使用GridSearchCV与间隔

时间:2017-12-24 15:22:11

标签: python machine-learning data-science

在python中使用Grid搜索分类器时使用这个函数GridSearchCV()想象我们有一个参数间隔来调整表单1到100我们如何指定它(1:100不起作用)??

1 个答案:

答案 0 :(得分:1)

这是你应该如何将参数传递给GridSearchCV()函数:

pipeline = Pipeline([
    ('clf', OneVsRestClassifier(SVC(), n_jobs=1)),
])

parameters = [

    {'clf__estimator__kernel': ['rbf'],
     'clf__estimator__gamma': [1e-3, 1e-4],
     'clf__estimator__C': [1, 10]
    },

    {'clf__estimator__kernel': ['poly'],
     'clf__estimator__C': [1, 10]
    }
     ]

grid_search_tune = GridSearchCV(pipeline, parameters, cv=2, n_jobs=3, verbose=10)