1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > knn鸢尾花数据集java_机器学习——鸢尾花数据集(Knn分类)

knn鸢尾花数据集java_机器学习——鸢尾花数据集(Knn分类)

时间:2019-12-25 16:52:22

相关推荐

knn鸢尾花数据集java_机器学习——鸢尾花数据集(Knn分类)

Iris数据集是常用的分类实验数据集,由Fisher, 1936收集整理。Iris也称鸢尾花卉数据集,是一类多重变量分析的数据集。数据集包含150个数据样本,分为3类,每类50个数据,每个数据包含4个属性。可通过花萼长度,花萼宽度,花瓣长度,花瓣宽度4个属性预测鸢尾花卉属于(Setosa,Versicolour,Virginica)三个种类中的哪一类。

iris以鸢尾花的特征作为数据来源,常用在分类操作中。该数据集由3种不同类型的鸢尾花的各50个样本数据构成。其中的一个种类与另外两个种类是线性可分离的,后两个种类是非线性可分离的。

该数据集包含了4个属性:

& Sepal.Length(花萼长度),单位是cm;

& Sepal.Width(花萼宽度),单位是cm;

& Petal.Length(花瓣长度),单位是cm;

& Petal.Width(花瓣宽度),单位是cm;

1.导包

import numpy as np

from sklearn import datasets

from sklearn.neighbors import KNeighborsClassifier

from sklearn.model_selection import train_test_split

2.数据集导入

①from sklearn import datasets

iris = datasets.load_iris()

②提前下载好数据集,用pandas包导入

例:iris=pd.read_csv('文件名',columns=['列名'])

3.查看数据特征

iris.data.shape

4. 拆分属性数据

iris_X = iris.data

5.# 拆分类别数据

iris_y = iris.target

6.预测数据

#准备训练集和测试集

iris_train_X , iris_test_X, iris_train_y ,iris_test_y = train_test_split(iris_X, iris_y, test_size=0.2,random_state=0)

#创建模型

knn = KNeighborsClassifier(n_neighbors=6)

#训练模型

knn.fit(iris_train_X, iris_train_y)

#预测

predict_result=knn.predict(iris_test_X)

print('预测结果',predict_result)

# 计算预测的准确率

print('预测准确率',knn.score(iris_test_X, iris_test_y))

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。