手记

机器学习-kNN实现简单的手写数字识别系统

功能


利用k-邻近算法,实现识别数字0到9

开发环境


数据集和项目源代码


Github:https://github.com/jiangkang/KMachineLearning.git

数据集


下面是32*32的黑白图像

32* 32像素数据集

为了方便处理,提供了文本文件

文本数据

数据集分为训练集和测试集,其中训练集是已经分类好的数据,测试集是用来测试算法。

将数据转换为特征向量


从上图可知,拿到的是32*32的的矩阵,每一个点是一个像素值,将这1024(32
32)个数值,转换成(1,1024)的向量。

数据转换为特征向量

KNN分类器


参考这篇文章机器学习-从kNN开始

kNN分类器

输入测试集,测试算法


测试算法

看看选择不同的k值,分类效果如何


取不同的k值

这里看错误的个数,绘制一个折线图
注意:执行时间比较长,去喝杯咖啡吧

这里取1-4,当然你也可以取更高的k值(速度慢)

实验结果证明,取k=3,效果比较好。

txt转换为png图像


为了更具真实性,同样写了txt转换为png的代码。

txt转png

图像转txt文件


image.png

总结


数据集如果稍大一点,执行时间会变得非常长



作者:姜康
链接:https://www.jianshu.com/p/a81ce10c4076

0人推荐
随时随地看视频
慕课网APP