GridSearch在OneVsRestClassifier中查找一个估算器

GridSearch在OneVsRestClassifier中查找一个估算器,第1张

GridSearch在OneVsRestClassifier中查找一个估算器

当您将嵌套估计量与网格搜索一起使用时,可以将参数的范围

__
作为分隔符。在这种情况下,SVC模型存储为
estimator
OneVsRestClassifier
模型内部命名的属性:

from sklearn.datasets import load_irisfrom sklearn.multiclass import oneVsRestClassifierfrom sklearn.svm import SVCfrom sklearn.grid_search import GridSearchCVfrom sklearn.metrics import f1_scoreiris = load_iris()model_to_set = oneVsRestClassifier(SVC(kernel="poly"))parameters = {    "estimator__C": [1,2,4,8],    "estimator__kernel": ["poly","rbf"],    "estimator__degree":[1, 2, 3, 4],}model_tunning = GridSearchCV(model_to_set, param_grid=parameters,       score_func=f1_score)model_tunning.fit(iris.data, iris.target)print model_tunning.best_score_print model_tunning.best_params_

产生:

0.973290762737{'estimator__kernel': 'poly', 'estimator__C': 1, 'estimator__degree': 2}


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5013903.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-11-14
下一篇 2022-11-15

发表评论

登录后才能评论

评论列表(0条)

保存