深度学习案例（三）：Mnist数据集分类简单案例

最新推荐文章于 2023-11-01 21:10:52 发布

德雅路王吸葱

最新推荐文章于 2023-11-01 21:10:52 发布

阅读量755

点赞数

文章标签： tensorflow Mnist

本文链接：https://blog.csdn.net/weixin_42174832/article/details/80422310

版权

本文介绍了使用TensorFlow进行MNIST数据集的分类案例，数据集来源于Yann LeCun的网站，下载后直接用于项目。

摘要由CSDN通过智能技术生成

MNIST数据集的官网是Yann LeCun’s website。下载数据包后放入工程目录下即可。

代码如下

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data

# 载入数据集
mnist = input_data.read_data_sets("MNIST_data", one_hot=True)

# 每个批次的大小
batch_size = 100
# 计算一共有多少个批次
n_batch = mnist.train.num_examples // batch_size

# 定义两个placeholder
x = tf.placeholder(tf.float32, [None, 784])
y = tf.placeholder(tf.float32, [None, 10])

#