
适合初学者的机器学习和深度学习小项目,涵盖随机森林、多层感知机及卷积神经网络,并实现训练可视化与多分类混淆矩阵等功能。
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本教程为初学者设计,包含随机森林、多层感知机和卷积神经网络等基础项目的实践,同时提供训练可视化工具和多分类混淆矩阵的实现方法。
这是一个适合新手入门的机器学习与深度学习项目,涵盖了随机森林、多层感知器及卷积神经网络等内容,并实现了模型训练、验证、推理等功能,同时支持训练过程可视化以及特征图展示。
所需环境:
- scikit-learn
- pytorch, torchvision
- opencv
- pillow
- tensorboardx
使用步骤:
对于深度学习部分,
1. 将项目下载到本地。
2. 训练模型:运行deep_learning.py,程序会自动下载fashion mnist数据集并开始训练。可以选择使用CNN或MLP进行训练。
3. 可视化训练过程:在上一步完成后,在当前目录下将自动生成runs文件夹(包含tensorboard记录的训练信息)和output文件夹(存储输出模型)。通过命令行执行`tensorboard --logdir=.runs`,然后打开浏览器访问生成的网址即可查看loss曲线及网络结构可视化。
4. 模型推理:运行infer.py进行预测操作。
5. 打印验证集混淆矩阵:运行mcm.py以显示分类效果评估。
对于机器学习部分:
1. 运行machine_learning.py,程序将输出分类信息和多类别的混淆矩阵。
全部评论 (0)
还没有任何评论哟~


