python数字识别bayes_OpenBayes 下识别手写数字

本文通过 MNIST 数据集,介绍如何在 OpenBayes 平台上使用 Pytorch 和 Tensorflow 进行手写数字识别模型的训练与推断。详细步骤包括代码获取、任务创建、模型训练以及模型评估,展示了两种框架在图像分类任务中的应用。
摘要由CSDN通过智能技术生成

了解过机器学习,尤其是最近几年大热的深度学习方法的朋友很可能有听过过 MNIST 这个数据集,数据集来自美国国家标准与技术研究所,National Institute of Standards and Technology (NIST)。训练集(training set)由来自 250 个不同人手写的数字构成,测试集(test set)也是同样比例的手写数字数据。

我们这里就借用这个数据集来介绍在 OpenBayes 中采用 Pytorch / Tensorflow 进行图片的分类。其中会使用到 OpenBayes 的数据集绑定、模型的训练以及模型的使用。

Tensorflow

获取代码

从 Github 下载样例代码:

git clone https://github.com/signcl/openbayes-mnist-example.git

cd openbayes-mnist-example/tensorflow

切换到所下载代码目录看到代码结构如下:

.

├── inference.py

├── openbayes.yaml

└── train.py

其中, .py 包含了执行机器学习模型训练和采用模型进行推断的代码

openbayes.yaml 包含了执行一个「Python 脚本」的配置,更多信息见 openbayes cli 配置文件

创建第一个任务 注意 这里介绍的是通过页面上传代码压缩包的流程,但是这已经不再是我们推荐的创建「Python 脚本」的最佳实践了,更好的方式参见 bayes 命令行工具入门。

数据集准备好后在页面左侧导航栏点击「新建算力容器」创建一个 Python 脚本执行的

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值