博客
关于我
SKLearn中SVM参数自动选择的最简单示例(使用GridSearchCV)
阅读量:580 次
发布时间:2019-03-11

本文共 980 字,大约阅读时间需要 3 分钟。

大家都知道,SVM在调参合适的情况下可以表现出很好的分类效果,但其参数较多,尤其是在这里提到的几个重要参数C、gamma和kernel的选择上,调整对模型的表现会有显著影响。一些人对SVM的调参过程做了详细的解释,而对于SKLearn来说,可以采用GridSearchCV来自动化地进行参数搜索,有朋友对它的使用方法进行了详细的介绍。

网友提到,在使用GridSearchCV进行SVM调参时,主要需要重点关注kernel、C和gamma这三个参数的选择。在提供的代码示例中,可以看到如何通过GridSearchCV对这些参数进行自动搜索。该代码示例适用于分类任务中使用SVM的场景。

在使用GridSearchCV时,可以指定多个候选值进行搜索,例如kernel的候选值可以是线性或rbf两种,C的候选值可以选择1、2、4等,gamma的候选值可以选择0.125、0.25等。需要注意的是,这些参数的选择会对最终模型的性能产生重要影响。

关于scoring参数,在GridSearchCV中可以指定不同的评分函数。在分类任务中常用的评分函数包括accuracy、average_precision、f1、f1_micro等。例如,'f1'评分函数适用于二分类任务,可以综合考虑准确率、召回率和调整率的平均值。

以下是我们提供的一个最简单的示例代码:

from sklearn import svmfrom sklearn.model_selection import GridSearchCVsvr = svm.SVC()parameters = {    'kernel': ['linear', 'rbf'],    'C': [1, 2, 4],    'gamma': [0.125, 0.25, 0.5, 1, 2, 4]}clf = GridSearchCV(svr, parameters, scoring='f1')clf.fit(X, y)print('The parameters of the best model are: ')print(clf.best_params_)

在使用GridSearchCV时,可以灵活地选择搜索的参数范围,并通过scoring参数指定评分函数。此外,'f1'评分函数可以有效地衡量分类模型的综合性能。

转载地址:http://zhstz.baihongyu.com/

你可能感兴趣的文章
Frame--WEB框架
查看>>
idea 在Debug 模式中运行语句中函数的方法
查看>>
eclipse“SVN检出”遇到问题 error getting dir list 的解决办法
查看>>
springboot2.1.1开启druid数据库连接池并开启监控
查看>>
vscode bash-4.3$ bash:git: command not found问题处理
查看>>
《朝花夕拾》金句摘抄(五)
查看>>
《朝花夕拾》金句摘抄(六)
查看>>
Boostrap技能点整理之【网格系统】
查看>>
新闻发布项目——业务逻辑层(UserService)
查看>>
新闻发布项目——后台JSP界面adminManage/modifyCategory.jsp
查看>>
hibernate正向生成数据库表以及配置——hibernate.cfg.xml
查看>>
javaWeb服务详解(含源代码,测试通过,注释) ——Emp的Dao层
查看>>
java实现人脸识别源码【含测试效果图】——Dao层(IUserDao)
查看>>
使用ueditor实现多图片上传案例——前台数据层(Index.jsp)
查看>>
ssh(Spring+Spring mvc+hibernate)——Dept.hbm.xml
查看>>
ssh(Spring+Spring mvc+hibernate)——updateEmp.jsp
查看>>
ssm(Spring+Spring mvc+mybatis)——saveDept.jsp
查看>>
ssm(Spring+Spring mvc+mybatis)——updateDept.jsp
查看>>
JavaScript操作BOM对象
查看>>
解决Chrome播放视频闪屏黑屏无法播放
查看>>