计算机视觉学习之路之tensorflow学习-02程序实现鸢尾花数据集分类

数据集回顾

先回顾鸢尾花数据集，其提供了 150 组鸢尾花数据，每组包括鸢尾花的花萼长、花萼宽、花瓣长、花瓣宽 4 个输入特征，同时还给出了这一组特征对应的鸢尾花类别。类别包括狗尾鸢尾、杂色鸢尾、弗吉尼亚鸢尾三类， 分别用数字0、1、2 表示。使用此数据集代码如下：

from sklearn.datasets import load_iris
x_data = datasets.load_iris().data # 返回 iris 数据集所有输入特征
y_data = datasets.load_iris().target # 返回 iris 数据集所有标签

即从 sklearn 包中导出数据集，将输入特征赋值给 x_data 变量，将对应标签赋值给 y_data 变量。

程序实现

我们用神经网络实现鸢尾花分类仅需要三步：
(1)准备数据，包括数据集读入、数据集乱序，把训练集和测试集中的数据配成输入特征和标签对，生成 train 和 test 即永不相见的训练集和测试集；
(2)搭建网络，定义神经网络中的所有可训练参数；
(3)优化这些可训练的参数，利用嵌套循环在 with 结构中求得损失函数 loss对每个可训练参数的偏导数，更改这些可训练参数，为了查看效果，程序中可以加入每遍历一次数据集显示当前准确率，还可以画出准确率 acc 和损失函数 loss的变化曲线图。

以上部分的完整代码与解析如下：
(1) 数据集读入：

from sklearn.datasets import datasets
x_data = datasets.load_iris().data # 返回 iris 数据集所有输入特征
y_data = datasets.load_iris().target # 返回 iris 数据集所有标签

(2) 数据集乱序：

np.random.seed(116) # 使用相同的 seed，使输入特征/标签一一对应
np.random.shuffle(x_data)
np.random.seed(116)
np.random.shuffle(y_data) 
tf.random.set_seed(116)

(3) 数据集分割成永不相见的训练集和测试集：

x_train = x_data[:-30] #截取从0到倒数第30个数据为训练集
y_train = y_data[:-30