
fashion MNIST训练
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Fashion MNIST训练是指使用包含衣物图像数据集进行机器学习模型训练的过程,旨在提升模型在时尚物品分类上的识别精度。
fashion-mnist-train详细测试结果参考目录下的fashion_mnist.docx文件。本项目使用了三种机器学习分类器:随机森林、KNN和朴素贝叶斯,以及一个深度学习卷积网络结构(包含两层卷积层和一层全连接层)。数据集为fashion-mnist。
使用的工具包括sk-learn进行机器学习模型训练,并用pytorch在GPU上运行深度学习框架。以下是代码的运行流程:
1. 获取数据集:解压位于fashion_mnist目录下的fashion_mnist_data.zip,该文件包含四个文件,分别为训练集、测试集以及对应的标签。
2. 数据可视化和标签制作:执行make_data.py脚本,在fashion_mnist下生成训练集与测试集中图片的文件,并创建相应的标签数据。
3. 机器学习分类器性能评估:通过运行train_minst.py可以分别对三种不同类型的机器学习分类器进行性能测试。
4. 深度学习卷积网络模型验证:执行fashion_mnist_cnn.py脚本,可以通过改变超参数L来调整深度学习模型。
全部评论 (0)
还没有任何评论哟~


