TensorFlow训练MNIST数据集提示参数未初始化

在运行TensorFlow的相关代码时出现错误

环境:python3.7
tensorflow1.14.0

'''
该部分内容为TensorFlow进阶指南
32页内容
时间:2021年3月30日
训练MNIST数据集
'''

import LeNet_5.input_data as inputdata
import tensorflow as tf

sess=tf.compat.v1.Session()
mnist=inputdata.read_data_sets('MNIST',one_hot=True)
# 一个等待输入的结点,将这个节点设置为float,并设定形状为【None]
x=tf.placeholder("float",[None,784])
#用Variable定义两个变量
w=tf.Variable(tf.zeros([784,10]))
b=tf.Variable(tf.zeros([10]))
#计算softmax值
y=tf.nn.softmax(tf.matmul(x,w)+b)
#一个等待输入的节点。。。
y_=tf.compat.v1.placeholder("float",[None,10])
#定义一个交叉熵
cross_entropy=-tf.reduce_sum(y_*tf.math.log(y))
#梯度下降法创建一个优化器
train_step=tf.compat.v1.train.GradientDescentOptimizer(0.01).minimize(cross_entropy)
#使用循环进行训练
init=tf.compat.v1.global_variables_initializer()
for i in range(3000):
    batch_xs,batch_ys=mnist.train.next_batch(100)
    feed_dict={x : batch_xs,y_ : batch_ys}

    sess.run(train_step,feed_dict)
#评估训练的精度如何
correct_prediction=tf.equal(tf.arg_max(y,1),tf.arg_max(y_,1))
accuracy=tf.reduce_mean(tf.cast(correct_prediction,"float"))
#输出测试数据集的正确率
print(sess.run(accuracy,feed_dict={x:mnist.test.images,y_:mnist.test.labels}))

错误如下:

tensorflow.python.framework.errors_impl.FailedPreconditionError: Attempting to use uninitialized value Variable

通过错误描述可以看出原因是没有初始化参数,但是明明有init=tf.compat.v1.global_variables_initializer()初始化的语句,最后发现是少了一句sess.run(init),注意不要放错位置,正确代码如下

'''
该部分内容为TensorFlow进阶指南
32页内容
时间:2021年3月30日
训练MNIST数据集
'''

import LeNet_5.input_data as inputdata
import tensorflow as tf

sess=tf.compat.v1.Session()
mnist=inputdata.read_data_sets('MNIST',one_hot=True)
# 一个等待输入的结点,将这个节点设置为float,并设定形状为【None]
x=tf.placeholder("float",[None,784])
#用Variable定义两个变量
w=tf.Variable(tf.zeros([784,10]))
b=tf.Variable(tf.zeros([10]))
#计算softmax值
y=tf.nn.softmax(tf.matmul(x,w)+b)
#一个等待输入的节点。。。
y_=tf.compat.v1.placeholder("float",[None,10])
#定义一个交叉熵
cross_entropy=-tf.reduce_sum(y_*tf.math.log(y))
#梯度下降法创建一个优化器
train_step=tf.compat.v1.train.GradientDescentOptimizer(0.01).minimize(cross_entropy)
#使用循环进行训练
init=tf.compat.v1.global_variables_initializer()
sess.run(init)
for i in range(3000):
    batch_xs,batch_ys=mnist.train.next_batch(100)
    feed_dict={x : batch_xs,y_ : batch_ys}

    sess.run(train_step,feed_dict)
#评估训练的精度如何
correct_prediction=tf.equal(tf.arg_max(y,1),tf.arg_max(y_,1))
accuracy=tf.reduce_mean(tf.cast(correct_prediction,"float"))
#输出测试数据集的正确率
print(sess.run(accuracy,feed_dict={x:mnist.test.images,y_:mnist.test.labels}))

运行结果如下:

0.9214

如果对你有帮助,点个赞哟😊😊😊

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是一个使用 TensorFlow 实现 MNIST 数据集的两个隐藏层全连接神经网络的示例代码: ``` import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data # 载入 MNIST 数据集 mnist = input_data.read_data_sets('MNIST_data', one_hot=True) # 定义网络参数 n_input = 784 # 输入层神经元数量(28*28) n_hidden_1 = 256 # 第一个隐含层神经元数量 n_hidden_2 = 256 # 第二个隐含层神经元数量 n_classes = 10 # 输出层神经元数量(0~9) # 定义占位符 x = tf.placeholder(tf.float32, [None, n_input]) y = tf.placeholder(tf.float32, [None, n_classes]) # 定义权重和偏置项 weights = { 'h1': tf.Variable(tf.random_normal([n_input, n_hidden_1])), 'h2': tf.Variable(tf.random_normal([n_hidden_1, n_hidden_2])), 'out': tf.Variable(tf.random_normal([n_hidden_2, n_classes])) } biases = { 'b1': tf.Variable(tf.random_normal([n_hidden_1])), 'b2': tf.Variable(tf.random_normal([n_hidden_2])), 'out': tf.Variable(tf.random_normal([n_classes])) } # 构建模型 def multilayer_perceptron(x): layer_1 = tf.add(tf.matmul(x, weights['h1']), biases['b1']) layer_1 = tf.nn.relu(layer_1) layer_2 = tf.add(tf.matmul(layer_1, weights['h2']), biases['b2']) layer_2 = tf.nn.relu(layer_2) out_layer = tf.matmul(layer_2, weights['out']) + biases['out'] return out_layer # 定义损失函数和优化器 pred = multilayer_perceptron(x) cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y)) optimizer = tf.train.AdamOptimizer(learning_rate=0.001).minimize(cost) # 初始化变量 init = tf.global_variables_initializer() # 训练模型 with tf.Session() as sess: sess.run(init) for epoch in range(100): avg_cost = 0. total_batch = int(mnist.train.num_examples / 100) for i in range(total_batch): batch_x, batch_y = mnist.train.next_batch(100) _, c = sess.run([optimizer, cost], feed_dict={x: batch_x, y: batch_y}) avg_cost += c / total_batch if epoch % 10 == 0: print("Epoch:", '%04d' % (epoch+1), "cost=", "{:.9f}".format(avg_cost)) print("Optimization Finished!") # 测试模型 correct_prediction = tf.equal(tf.argmax(pred, 1), tf.argmax(y, 1)) accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32)) print("Accuracy:", accuracy.eval({x: mnist.test.images, y: mnist.test.labels})) ``` 这段代码定义了一个包含两个隐含层的全连接神经网络,使用 ReLU 激活函数,交叉熵作为损失函数,Adam 优化器进行优化,最终输出模型的准确率。在训练过程中,每个 epoch 会遍历整个训练集,使用批量梯度下降进行参数更新。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值