基于SVM、Pipeline、GridSearchCV的鸢尾花分类

SVM中文叫做支持向量机,support vector machine的简写,是常用的分类方法。
Pipeline中文叫做管道,是sklearn中用来打包数据预处理、模型训练这2个步骤的常用方法。
GridSearchCV中文叫做交叉验证网格搜索,是sklearn库中用来搜索模型最优参数的常用方法。

2018年8月26日笔记

1.数据集

Iris(鸢尾花)数据集是多重变量分析的数据集。
数据集包含150行数据,分为3类,每类50行数据。
每行数据包括4个属性:Sepal Length(花萼长度)、Sepal Width(花萼宽度)、Petal Length(花瓣长度)、Petal Width(花瓣宽度)。可通过这4个属性预测鸢尾花属于3个种类的哪一类。
样本数据局部截图:

10345471-87df8234105ff9c7.png
Iris.png

获取150个样本数据的3种方法:
1.完整的样本数据Iris.csv文件下载链接: https://pan.baidu.com/s/16N0ivGWFrmc73ustPLWPZA 密码: ugun
2.数据集链接: https://gist.github.com/curran/a08a1080b88344b0c8a7
3.在sklearn的skleanrn库中自带了iris数据集,可以导入sklearn库的datasets文件,调用datasets文件中的load_iris方法就可以获得iris数据集。
本文采用的是第3种方法,直接从sklearn库中获取数据。

2.观察数据

  • 2
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
鸢尾花分类是机器学习中的典型问题之一,可以使用SVM(支持向量机)进行预测。SVM是一种二元分类方法,可以将样本分为两类,并找到最优的分割超平面。在鸢尾花分类中,我们需要将鸢尾花分为三类:山鸢尾、变色鸢尾和维吉尼亚鸢尾。 下面是基于SVM鸢尾花分类方法的步骤: 1. 导入数据集 首先,需要导入鸢尾花数据集,该数据集包含150个样本和4个特征(花萼长度、花萼宽度、花瓣长度和花瓣宽度),每个样本属于三个不同的鸢尾花品种之一。 2. 数据预处理 对于SVM模型,输入数据需要进行标准化处理。可以使用sklearn中的StandardScaler函数来进行标准化处理。 3. 拆分数据集 将数据集分成训练集和测试集,一般采用70%的数据作为训练集,30%的数据作为测试集。 4. 训练模型 使用sklearn中的SVM模型进行训练,可以采用不同的核函数(如线性核函数、多项式核函数、径向基函数等)。 5. 模型评估 使用测试集来评估模型的准确性,可以计算出模型的准确率、精确率、召回率和F1值。 下面是Python代码示例: ```python from sklearn import datasets from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler from sklearn.svm import SVC from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score # 导入鸢尾花数据集 iris = datasets.load_iris() X = iris.data y = iris.target # 数据标准化 scaler = StandardScaler() X_std = scaler.fit_transform(X) # 拆分数据集 X_train, X_test, y_train, y_test = train_test_split(X_std, y, test_size=0.3, random_state=0) # 训练模型 svm = SVC(kernel='linear', C=1.0, random_state=0) svm.fit(X_train, y_train) # 预测测试集 y_pred = svm.predict(X_test) # 模型评估 print('Accuracy: %.2f' % accuracy_score(y_test, y_pred)) print('Precision: %.2f' % precision_score(y_test, y_pred, average='macro')) print('Recall: %.2f' % recall_score(y_test, y_pred, average='macro')) print('F1: %.2f' % f1_score(y_test, y_pred, average='macro')) ``` 输出结果如下: ``` Accuracy: 0.98 Precision: 0.98 Recall: 0.98 F1: 0.98 ``` 可以看出,基于SVM鸢尾花分类方法可以很好地预测鸢尾花的品种,具有较高的准确率、精确率、召回率和F1值。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值