Python 机器学习简单实例：KNN预测鸢尾花分类

读研太忙了才发现一年没更新，就以当初学习机器学习的一个简单例子作为一个新的开始吧。以下为正文。

一、安装sklearn库

执行以下命令安装，注意：在此之前需要提前安装numpy、matplotlib、scipy库

pip install scikit-learn

如果速度较慢可更换国内镜像源，例如清华镜像源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

二、实例代码

1、引入所需模块

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier

2、读入数据

x_data = load_iris().data # 特征值
y_data = load_iris().target # 目标值

x_data代表每个样本有4个特征值，对应[花萼长]，[花萼宽]，[花瓣长]，[花瓣宽]。

y_data代表每个样本对应的目标值，共有3个类别。

3、划分数据集

x_train, x_test, y_train, y_test = train_test_split(x_data, y_data, test_size=0.2, random_state=10)

按照测试集占总样本数20%的比例划分训练集和测试集

4、特征标准化

scaler = StandardScaler()
x_train = scaler.fit_transform(x_train)
x_test = scaler.transform(x_test)

5、训练KNN模型

estimator = KNeighborsClassifier(n_neighbors=5) # 构建KNN模型
estimator.fit(x_train, y_train) # 训练模型

对于每个样本来说，取最邻近的5个样本作为当前样本分类的划分依据

6、模型评估

y_pre = estimator.predict(x_test) # 预测值
score = estimator.score(x_test, y_test) # 准确率
print("实际结果为：\n", y_test)
print("预测结果为：\n", y_pre)
print("对比结果为：\n", y_pre == y_test)
print("准确率为：\n", score)

运行结果为：

Python 机器学习简单实例：KNN预测鸢尾花分类相关推荐

[转载] Python 机器学习经典实例
参考链接: Python中的逻辑门内容介绍在如今这个处处以数据驱动的世界中,机器学习正变得越来越大众化.它已经被广泛地应用于不同领域,如搜索引擎.机器人.无人驾驶汽车等.本书首先通过实用的案例介绍 ...
python泰坦尼克号数据预测_机器学习入门之Python机器学习：泰坦尼克号获救预测一...
本文主要向大家介绍了机器学习入门之Python机器学习:泰坦尼克号获救预测一,通过具体的内容向大家展现,希望对大家学习机器学习入门有所帮助. 一.项目概要 1.应用模式识别.数据挖掘(核心).统计学 ...
机器学习从蛋白序列预测蛋白分类（二）
半路出家学机器学习,先在自己熟悉的领域尝试,每天进步一点点,记录自己成长过程(正式开始前的小叨叨) 接([机器学习从蛋白序列预测蛋白分类(一)])继续分析三,特征提取其实在(机器学习从蛋白序列预测 ...
KNN实现鸢尾花分类--数据可视化--根据花萼及花瓣分类--sklearn
KNN实现鸢尾花分类一.导入相关库二.导入数据 1.关于鸢尾花的数据 2.删除Id(不需要的)数据 3.分别利用花萼和花瓣的特征三.训练集及测试集 1.花瓣 2.花萼四.KNN算法 1.花瓣 ...
（机器学习）——使用KNN进行水果分类
(机器学习)--使用KNN进行水果分类(详解) 实验目标是:导入素材文件fruit_data.txt,使用SKlearn中的neighbors模块对水果数据进行分类,然后预测A.B两种水果的类别.以下 ...
《python机器学习经典实例》Expected 2D array, got 1D array instead和Reshape your data either using array.问题（已解决）
问题描述: ValueError: Expected 2D array, got 1D array instead: array=[2. 1.5]. Reshape your data either ...
[python机器学习]机器学习简单示例-KNN、决策树、线性回归、逻辑回归
1.KNN 查找距离已知的几个点最近的类型,并返回这个类型进行预测. 如小明在北京,小红在北京,小刚在河南,而我距离小明和小红比小刚近,则我最可能在北京而不是河南 #!/usr/bin/env pyt ...
【笔记】【机器学习基础】1.7 鸢尾花分类
1.7 鸢尾花分类问题背景:鸢尾花有三个品种 1.setosa 2.versicolor 3.virginica 目标:构建模型,从已知数据中预测新的鸢尾花的品种(三分类问题) 监督学习问题:从已知 ...
基于简单神经网络模型的鸢尾花分类问题
摘要本人在学习<Python机器学习基础教程>时的一些小实验. 一.认识鸢尾花数据 python的机器学习库scikit-learn中保存了大量的经典的实验数据集,在学习阶段没有办法搜集 ...