栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

More than one estimator in GridSearchCV(sklearn)

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

More than one estimator in GridSearchCV(sklearn)

GridSearchCV works on parameters. It will train multiple estimators (but same
class (one of SVC, or DecisionTreeClassifier, or other classifiers) with
different parameter combinations from specified in

param_grid
.
best_estimator_
is the estimator which performs best on the data.

So essentially

best_estimator_
is the same class object initialized with
best found params.

So in the basic setup you cannot use multiple estimators in the grid-search.

But as a workaround, you can have multiple estimators when using a pipeline in
which the estimator is a

"parameter"
which the GridSearchCV can set.

Something like this:

from sklearn.pipeline import Pipelinefrom sklearn.svm import SVCfrom sklearn.tree import DecisionTreeClassifierfrom sklearn.model_selection import GridSearchCVfrom sklearn.datasets import load_irisiris_data = load_iris()X, y = iris_data.data, iris_data.target# Just initialize the pipeline with any estimator you like    pipe = Pipeline(steps=[('estimator', SVC())])# Add a dict of estimator and estimator related parameters in this listparams_grid = [{     'estimator':[SVC()],     'estimator__C': [1, 10, 100, 1000],     'estimator__gamma': [0.001, 0.0001],     },     {     'estimator': [DecisionTreeClassifier()],     'estimator__max_depth': [1,2,3,4,5],     'estimator__max_features': [None, "auto", "sqrt", "log2"],     },    # {'estimator':[Any_other_estimator_you_want],    #  'estimator__valid_param_of_your_estimator':[valid_values]   ]grid = GridSearchCV(pipe, params_grid)

You can add as many dicts inside the list of

params_grid
as you like, but
make sure that each dict have compatible parameters related to the
'estimator'
.



转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/410763.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号