数据挖掘入门系列教程(八)之使用神经网络(基于pybrain)识别数字手写集MNIST

数据挖掘入门系列教程(八)之使用神经网络(基于pybrain)识别数字手写集MNIST

在本章节中,并不会对神经网络进行介绍,因此如果不了解神经网络的话,强烈推荐先去看《西瓜书》,或者看一下我的上一篇博客:数据挖掘入门系列教程(七点五)之神经网络介绍

本来是打算按照《Python数据挖掘入门与实践》里面的步骤使用神经网络来识别验证码,但是呢,验证码要自己生成,然后我又想了一下,不是有大名鼎鼎的MNIST数据集吗,为什么不使用它呢,他不香吗?

MNIST(Mixed National Institute of Standards and Technology database)相信大家基本上都了解过他,大部分的机器学习入门项目就是它。它是一个非常庞大的手写数字数据集(官网)。里面包含了0~9的手写的数字。部分数据如下:

数据集分为两个部分,训练集和测试集。然后在不同的集合中分为两个文件,数据Images文件和Labels文件。在数据集中一个有60,000个训练数据和10,000个测试数据。图片的大小是28*28。

下载数据集

万物始于数据集,尽管官网提供了数据集供我们下载,但是在sklearn中提供了更方便方法让我们下载数据集。代码如下:

import numpy as np  from sklearn.datasets import fetch_openml  # X为image数据,y为标签  X, y = fetch_openml('mnist_784', version=1, return_X_y=True)  

其中X,y中间既包含了训练集又包含了测试集。也就是说X或者y中有70,000条数据。 那么数据是什么呢?

在X中,每一条数据是一个长为(28 times 28=784)的数组,数组的数据是图片的像素值。每一条y数据就是一个标签,代表这张图片表示哪一个数字(从0到9)。

然后我们将数据进行二值化,像素值大于0的置为1,并将数据保存到文件夹中:

X[X > 0 ] = 1  np.save("./Data/dataset",X)  np.save("./Data/class",y)  

然后在Data文件夹中就出现了以下两个文件:

我们取出dataset中间的一条数据,然后转成28*28的格式,如下所示:

数据集既可以使用上面的方法得到,也可以从我的Github上面进行下载(其中dataset数据集因为GitHub文件大小的限制所以进行了压缩,需要解压才能够使用)。

加载数据集

前面的步骤我们下载好了数据集,现在我们就可以来加载数据了。

import numpy as np  X = np.load("./Data/dataset.npy")  y = np.load("./Data/class.npy")  

取出X中的一条数据如下所示:

取出y中的一条数据,如下所示:

一切都很完美,但是这里有一个问题,在神经网络中,输出层实际上是这样的:

它并不是直接输出某一个结果,而是输出(y_1,…,y_j,…,y_l)结果(在MNIST中(l=10),因为只有10种数字)。以上面的5为例子,输出层并不是单纯的输出只输出一个数字,而是要输出10个值。那么如何将输出5变成输出10个数字呢?这里我们使用”one hot Encoding“。

One-Hot编码,又称为一位有效编码,主要是采用(N)位状态寄存器来对(N)个状态进行编码,每个状态都由他独立的寄存器位,并且在任意时候只有一位有效。

以下面的数据为例,每一行代表一条数据,每一列代表一个属性。其中第2个属性只需要3个状态码,因为他只有0,1,2三种属性。这样我们就可以使用100代表0,010代表1,001代表2。

那么这个数据编码后的数据长什么样呢?如下图:

现在我们就可以将前面加载的数据集标签(y)进行“one hot Encoding”。

代码如下:

from sklearn.preprocessing import OneHotEncoder  # False代表不生成稀疏矩阵  onehot = OneHotEncoder(sparse = False)  # 首先将y转成行长为7000,列长为1的矩阵,然后再进行转化。  y = onehot.fit_transform(y.reshape(y.shape[0],1))  

接着就是切割数据集了。将数据集切割成训练集和测试集。

from sklearn.model_selection import train_test_split  x_train,x_test,y_train,y_test = train_test_split(X,y,random_state=14)  

在神经网络中,我们使用pybrain框架去构建一个神经网络。但是呢,对于pybrain库,他很与众不同,他要使用自己的数据集格式,因此,我们需要将数据转成它规定的格式。

from pybrain.datasets import SupervisedDataSet    train_data = SupervisedDataSet(x_train.shape[1],y.shape[1])  test_data = SupervisedDataSet(x_test.shape[1],y.shape[1])    for i in range(x_train.shape[0]):      train_data.addSample(x_train[i],y_train[i])  for i in range(x_test.shape[0]):      test_data.addSample(x_test[i],y_test[i])  

终于,数据集的加载就到这里结束了。接下来我们就可以开始构建一个神经网络了。

构建神经网络

首先我们来创建一个神经网络,网络中只含有输入层,输出层和一层隐层。

from pybrain.tools.shortcuts import buildNetwork  # X.shape[1]代表属性的个数,100代表隐层中神经元的个数,y.shape[1]代表输出  net = buildNetwork(X.shape[1],100, y.shape[1], bias=True)  

这里说有以下“bias”的作用。bias代表的是偏置神经元,bias = True代表偏置神经元激活,也就是在每一层都使用这个这个神经元。偏置神经元如下图,实际上也就是阈值,只不过换一种说法而已。

现在我们已经构建好了一个比较简单的神经网络,接下来我们就是使用BP算法去得到合适的权重值了。

反向传播(BP)算法

具体的算法步骤在上一篇博客已经介绍过了,很幸运的是在pybrain中间提供了BP算法的库供我们使用。这里就直接上代码吧。关于BackpropTrainer更加细节的使用可以看官网

from pybrain.supervised.trainers import BackpropTrainer  trainer = BackpropTrainer(net, train_data, learningrate=0.01,weightdecay=0.01)  

这里面有几个参数稍微的说明下:

  • net:神经网络

  • train_data:训练的数据集

  • learningrate:学习率,也就是下面的(eta),同样它可以使用lrdecay这个参数去控制衰减率,具体的就去看官网文档吧。

    [begin{equation}begin{array}{l} Delta w_{h j}=eta g_{j} b_{h} \ Delta theta_{j}=-eta g_{j} \ Delta v_{i h}=eta e_{h} x_{i} \ Delta gamma_{h}=-eta e_{h} \ end{array}end{equation} ]

  • weightdecay:权重衰减,权重衰减也就是下面的(lambda)

    [begin{equation} E=lambda frac{1}{m} sum_{k=1}^{m} E_{k}+(1-lambda) sum_{i} w_{i}^{2} \ lambda in(0,1) end{equation} ]

然后我们就可以开始训练了。

trainer.trainEpochs(epochs=100)  

epochs也就是训练集被训练遍历的次数。

接下载便是等待的时间了。等待训练集训练成完成。训练的时间跟训练集的大小,隐层神经元的个数,电脑的性能,步数等等有关。

切记切记,这一次的程序就不要在阿里云的学生机上面跑了,还是用自己的机器跑吧。尽管联想小新pro13 i5版本性能还可以,但是还是跑了一个世纪这么久,哎(耽误了我打游戏的时间)。

进行预测

通过前面的步骤以及等待一段时间后,我们就完成了模型的训练。然后我们就可以使用测试集进行预测。

predictions = trainer.testOnClassData(dataset=test_data)  

predictions的部分数据,代表着测试集预测的结果:

然后我们就可以开始验证准确度了,这里继续使用F1评估。这个已经在前面介绍过了,就不再介绍了。

F1验证

这里有个地方需要注意,因为之前的y_test数据我们使用one-hot encoding进行了编码,因此我们需要先将one-hot编码转成正常的形式。

# 取每一行最大值的索引。  y_test_arry = y_test.argmax(axis =1)  

具体效果如下:

然后使用F1值进行验证。

from sklearn.metrics import f1_score  print("F-score: {0:.2f}".format(f1_score(predictions,y_test_arry,average='micro')))  

然后结果如下:

结果只能说还行吧,不是特别的差,但是也不是特别的好。

总结

项目地址:Github。尽管上面的准确度不咋地,只有(86%),但是也还行吧,毕竟也是使用了一层隐层,然后隐层也只有100个神经元。

如果电脑的性能不够的话,可是适当的减少步数和训练集的大小,以及隐层神经元的个数。

参考