首页 > 其他 > 详细

3_鸢尾花

时间:2021-09-23 06:43:35      阅读:38      评论:0      收藏:0      [点我收藏+]

数据集

这是网上找的一个经典案例,很多算法都可以对其进行训练。

目前,有两种方式获取数据集

  • sklearn.datasets提供数据集接口
  • 通过pandas从网上导入

我将文档下载好,以csv的形式保存在pycharm上。文档内容如下:

技术分享图片

 

可以看出,第一行的前四项为鸢尾花特征,最后一项是鸢尾花类别,一共分三类。

导入pd包后,获取文档里面的数据,路径可填相对路径,并将数据集命名为df。

技术分享图片

 

通过keys查看属性,如果是sklearn.datasets里面的数据集,keys一般有data target等等

技术分享图片

 

 

 数据分割:前四项作为特征,最后一项做预测值

# %%
from sklearn.model_selection import train_test_split
X_train,X_test,y_train,y_test=train_test_split(df[df.keys()[0:4]],df[df.keys()[4]],test_size=0.25,random_state=33)

这里要注意一些细节,keys()是一个函数,要加括号。分割[0:4] 指的是从0开始到3(不包括4)

分割好后,输出一下:

技术分享图片

 

技术分享图片

 

 数据标准化

# 从sklearn.preprocessing里导入StandardScaler。
from sklearn.preprocessing import StandardScaler
ss = StandardScaler()
X_train = ss.fit_transform(X_train)
X_test = ss.transform(X_test)

标准化后的数据,标准化我们只对输入特征标准化,因为他们都是数值,可以量化。而类别是字符串,不能标准化。

技术分享图片

 

 采用逻辑斯特对其训练

lr = LogisticRegression()
# 调用LogisticRegression中的fit函数/模块用来训练模型参数。
lr.fit(X_train, y_train)
# 使用训练好的模型lr对X_test进行预测,结果储存在变量lr_y_predict中。
lr_y_predict = lr.predict(X_test)

输出测试集的结果

技术分享图片

 

 查看评分

技术分享图片

 

可见正确率还是挺高的。

自写一个测试

技术分享图片

 

3_鸢尾花

原文:https://www.cnblogs.com/ld66/p/15311867.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!