我关注tensorflow tutorial。最近的张量流更新已经修改了成本函数softmax_cross_entropy_with_logits()
。因此,本教程中的代码会出现以下错误:
ValueError: Only call softmax_cross_entropy_with_logits with named arguments (labels=..., logits=..., ...)
这是什么意思以及如何纠正它?
以下是整个代码:
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("/tmp/data/", one_hot = True)
n_nodes_hl1 = 500
n_nodes_hl2 = 500
n_nodes_hl3 = 500
n_classes = 10
batch_size = 100
x = tf.placeholder('float', [None, 784])
y = tf.placeholder('float')
def neural_network_model(data):
hidden_1_layer = {'weights':tf.Variable(tf.random_normal([784, n_nodes_hl1])),
'biases':tf.Variable(tf.random_normal([n_nodes_hl1]))}
hidden_2_layer = {'weights':tf.Variable(tf.random_normal([n_nodes_hl1, n_nodes_hl2])),
'biases':tf.Variable(tf.random_normal([n_nodes_hl2]))}
hidden_3_layer = {'weights':tf.Variable(tf.random_normal([n_nodes_hl2, n_nodes_hl3])),
'biases':tf.Variable(tf.random_normal([n_nodes_hl3]))}
output_layer = {'weights':tf.Variable(tf.random_normal([n_nodes_hl3, n_classes])),
'biases':tf.Variable(tf.random_normal([n_classes])),}
l1 = tf.add(tf.matmul(data,hidden_1_layer['weights']), hidden_1_layer['biases'])
l1 = tf.nn.relu(l1)
l2 = tf.add(tf.matmul(l1,hidden_2_layer['weights']), hidden_2_layer['biases'])
l2 = tf.nn.relu(l2)
l3 = tf.add(tf.matmul(l2,hidden_3_layer['weights']), hidden_3_layer['biases'])
l3 = tf.nn.relu(l3)
output = tf.matmul(l3,output_layer['weights']) + output_layer['biases']
return output
def train_neural_network(x):
prediction = neural_network_model(x)
cost = tf.reduce_mean( tf.nn.softmax_cross_entropy_with_logits(prediction,y) )
optimizer = tf.train.AdamOptimizer().minimize(cost)
答案 0 :(得分:37)
更改
tf.nn.softmax_cross_entropy_with_logits(prediction,y)
到
tf.nn.softmax_cross_entropy_with_logits(logits=prediction, labels=y)
答案 1 :(得分:3)
不推荐使用功能softmax_cross_entropy_with_logits
。新功能为softmax_cross_entropy_with_logits_v2
cost=tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(pred, y))
答案 2 :(得分:0)
这是因为代码不支持Tensorflow v 1.0
当您对以下代码行稍作更改时,它可以正常工作:
替换
cost = tf.reduce_mean( tf.nn.softmax_cross_entropy_with_logits(prediction,y) )
与
cost = tf.reduce_mean( tf.nn.softmax_cross_entropy_with_logits(logits=prediction, labels=y))