mnist softmax函数

快来打我* 2022-05-19 05:08 244阅读 0赞

# -*- coding: utf-8 -*-
    
    
    # 读取数据图片，预处理
    import tensorflow as tf
    import tensorflow.examples.tutorials.mnist.input_data as input_data  # 加载mnist数据
    mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)  # one_hot为是否将标签转为一维数组形式
    
    
    # 构建softmax回归模型
    sess = tf.InteractiveSession()  # 交互作用类（不需用在计算前构建整个图）
    
    
    # 占位符（x和y_）
    x = tf.placeholder("float", shape=[None, 784])  # 浮点数，二维数组（第一维大小不定，第二维是784）
    y_ = tf.placeholder("float", shape=[None, 10])  # 用于代表对应某一MNIST图片的类别
    
    
    # 变量（w权重和b偏置）
    w = tf.Variable(tf.zeros([784, 10]))  # 784*10的可变参数值二维矩阵
    b = tf.Variable(tf.zeros([10]))  # 10维的向量
    sess.run(tf.initialize_all_variables())  # 初始化所有变量为0
    
    
    # 类别预测与损失函数
    y = tf.nn.softmax(tf.matmul(x, w) + b)  # 计算每个分类的softmax概率值
    cross_entropy = -tf.reduce_sum(y_ * tf.log(y))  # 损失函数（目标类别和预测类别之间的交叉熵）
    
    
    # 训练模型
    train_step = tf.train.GradientDescentOptimizer(0.01).minimize(cross_entropy)  # 最速下降法，步长为0.01
    for i in range(1000):
        batch = mnist.train.next_batch(50)  # 每步加载50个样本
        train_step.run(feed_dict={
       x: batch[0], y_: batch[1]})  # feed_dict被每次训练的数据替代
    
    
    # 模型评估
    correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(y_, 1))  # 检测预测值与实际值是否匹配
    accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))  # 将布尔数组转化为正确率
    print(accuracy.eval(feed_dict={
       x: mnist.test.images, y_: mnist.test.labels}))  # 输出最终正确率

发表评论取消回复

表情：

评论列表（有 0 条评论，244人围观）

还没有评论，来说两句吧...

相关阅读

相关 softmax函数的改进

softmax函数的运算涉及到了指数函数的计算，故对于计算机而言就存在了一个 “溢出问题”。因为指数函数的增长容易变得非常大，这些超大值之间进行除法，结果会出现 nan（not

女爷i/ 2023年05月29日 13:17/ 0 赞/ 7 阅读

相关【Pytorch】Softmax激活函数

文章目录 Softmax 激活函数 Softmax 函数的偏导过程代码模拟偏导过程 Softmax 激活函数功能：不同的输入经过 so

迷南。/ 2022年11月05日 03:16/ 0 赞/ 296 阅读

相关 tensorflow(1) mnist_softmax.py

标签（空格分隔）：tensorflow -------------------- 1. 数据读取 `mnist = input_data.read_data_sets

小灰灰/ 2022年06月13日 09:51/ 0 赞/ 187 阅读

相关使用Softmax回归实现MNIST

小弟刚学习入门深度学习，首选tensorflow，不仅仅是因为他高效，热门，最主要的是他有很多资源可以查阅。为了尊敬知识，先声明本节代码参考《TensorFlow技术解析

亦凉/ 2022年06月04日 00:07/ 0 赞/ 274 阅读

相关 softmax函数

softmax函数 softmax用于多分类过程中，它将多个神经元的输出，映射到（0,1）区间内，可以看成概率来理解，从而来进行多分类！假设我们有一个数组V，Vi表示

桃扇骨/ 2022年05月23日 12:24/ 0 赞/ 276 阅读

相关 mnist softmax函数

-- coding: utf-8 -- 读取数据图片，预处理 import tensorflow as tf impor

快来打我*/ 2022年05月19日 05:08/ 0 赞/ 245 阅读

相关 MNIST softmax函数

使用softmax函数使用[mnist数据集][mnist] -- coding: utf-8 -- 读取数据图片，预处

傷城~/ 2022年05月19日 01:59/ 0 赞/ 195 阅读

相关 Python实现softmax函数

Python实现softmax函数： ![20190321174014558.png][] PS：为了避免求exp(x)出现溢出的情况，一般需要减去最大值。

旧城等待，/ 2022年03月15日 15:11/ 0 赞/ 410 阅读

相关 Pytorch softmax激活函数

可以用于将多个输出值转换成多个概率值，使每个值都符合概率的定义，范围在\[0, 1\]，且概率相加和为1，非常适合多分类问题。Softmax往往用在最后对输出值y的处理上。它会

小灰灰/ 2021年11月17日 01:24/ 0 赞/ 583 阅读

相关 softmax 函数

机器学习分为两种，一种是做分类，一种是做回归。softmax函数就是用来做分类的。 softmax 函数定义： ![20190813225013710.png][]

我就是我/ 2021年11月02日 13:08/ 0 赞/ 395 阅读