前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >机器学习中的参数调整

机器学习中的参数调整

作者头像
张俊红
发布2018-04-17 14:35:38
2.5K0
发布2018-04-17 14:35:38
举报
文章被收录于专栏:张俊红

总第102篇

前言

我们知道每个模型都有很多参数是可以调节的,比如SVM中使用什么样的核函数以及C值的大小,决策树中树的深度等。在特征选好、基础模型选好以后我们可以通过调整模型的这些参数来提高模型准确率。每个模型有很多参数,而每个参数又有很多不同的取值,我们该怎么调,最简单的一个方法就是一个一个试。sklearn中提供了这样的库代替了我们手动去试的过程,就是GridSearchCV,他会自己组合不同参数的取值,然后输出效果最好的一组参数。

GridSearchCV参数解释

代码语言:javascript
复制
GridSearchCV(estimator, param_grid, scoring=None, fit_params=None, \
             n_jobs=1, iid=True, refit=True, cv=None, \
             verbose=0, pre_dispatch=‘2*n_jobs’, error_score=’raise’, \
             return_train_score=’warn’)

estimator:所使用的基础模型,比如svc param_grid:是所需要的调整的参数,以字典或列表的形式表示 scoring:准确率评判标准 n_jobs:并行运算数量(核的数量 ),默认为1,如果设置为-1,则表示将电脑中的cpu全部用上 iid:假设数据在每个cv(折叠)中是相同分布的,损失最小化是每个样本的总损失,而不是折叠中的平均损失。 refit:默认为True,程序将会以交叉验证训练集得到的最佳参数,重新对所有可用的训练集与开发集进行,作为最终用于性能评估的最佳模型参数。 cv:交叉验证折叠数,默认是3,当estimator是分类器时默认使用StratifiedKFold交叉方法,其他问题则默认使用KFold verbose:日志冗长度,int类型,0:不输出训练过程,1:偶尔输出,>1:对每个子模型都输出 pre_dispatch:控制job数量,避免job过多出现内存错误

GridSearchCV对象

cv_results_:用来输出cv结果的,可以是字典形式也可以是numpy形式,还可以转换成DataFrame格式 best_estimator_:通过搜索参数得到的最好的估计器,当参数refit=False时该对象不可用 best_score_:float类型,输出最好的成绩 best_params_:通过网格搜索得到的score最好对应的参数

GridSearchCV方法

decision_function(X):返回决策函数值(比如svm中的决策距离) predict_proba(X):返回每个类别的概率值(有几类就返回几列值) predict(X):返回预测结果值(0/1) score(X, y=None):返回函数 get_params(deep=True):返回估计器的参数 fit(X,y=None,groups=None,fit_params):在数据集上运行所有的参数组合 transform(X):在X上使用训练好的参数

GridSearchCV实例

代码语言:javascript
复制
from sklearn import svm, datasets
from sklearn.model_selection import GridSearchCV
iris = datasets.load_iris()
parameters = {'kernel':('linear', 'rbf'), 'C':[1, 10]}
svc = svm.SVC()
clf = GridSearchCV(svc, parameters)
clf.fit(iris.data, iris.target)
------------------------------------------------------
GridSearchCV(cv=None, error_score='raise',
       estimator=SVC(C=1.0, cache_size=200, class_weight=None, coef0=0.0,
  decision_function_shape=None, degree=3, gamma='auto', kernel='rbf',
  max_iter=-1, probability=False, random_state=None, shrinking=True,
  tol=0.001, verbose=False),
       fit_params={}, iid=True, n_jobs=1,
       param_grid={'kernel': ('linear', 'rbf'), 'C': [1, 10]},
       pre_dispatch='2*n_jobs', refit=True, return_train_score=True,
       scoring=None, verbose=0)

clf.cv_results_

我们可以看到,cv_result一共有四组,这正是两个kernel和两个C的随机组合成四组。

clf.best_estimator_

clf.best_params_

clf.best_params_

注:本方法只适用于数据量较小的模型,不适合数据量过大的模型。

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2018-04-14,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 俊红的数据分析之路 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 前言
  • GridSearchCV参数解释
  • GridSearchCV对象
  • GridSearchCV方法
  • GridSearchCV实例
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档