☰

1 つのグリッド検索で複数の推定値を試す 質問する

tag-icon • tag-icon scikit-learn grid-search
1 つのグリッド検索で複数の推定値を試す 質問する

できる方法はあるでしょうかグリッド検索一度に複数の推定値スクリアンまたは他のライブラリ。たとえば、SVM とランダム フォレストを 1 つのグリッド検索で渡すことはできますか?

ベストアンサー1

はい。例:

pipeline = Pipeline([
    ('vect', CountVectorizer()),
    ('clf', SGDClassifier()),
])
parameters = [
    {
        'vect__max_df': (0.5, 0.75, 1.0),
        'clf': (SGDClassifier(),),
        'clf__alpha': (0.00001, 0.000001),
        'clf__penalty': ('l2', 'elasticnet'),
        'clf__n_iter': (10, 50, 80),
    }, {
        'vect__max_df': (0.5, 0.75, 1.0),
        'clf': (LinearSVC(),),
        'clf__C': (0.01, 0.5, 1.0)
    }
]
grid_search = GridSearchCV(pipeline, parameters)

おすすめ記事