4

次のモデルでは、アイデアを試してみたいと思っています。私のラベルは相互に排他的ではないので、特にtf.nn.sigmoid_cross_entropy_with_logits()に興味があります。tf.nn.sigmoid_cross_entropy_with_logits企業のドキュメントについての引数について

import tensorflow as tf 

from tensorflow.examples.tutorials.mnist import input_data 
mnist = input_data.read_data_sets('MNIST_data', one_hot=True) 

x = tf.placeholder(tf.float32, shape=[None, 784]) 
y_ = tf.placeholder(tf.float32, shape=[None, 10]) 

w1 = tf.get_variable("w1", shape=[784, 512], initializer=tf.contrib.layers.xavier_initializer()) 
b1 = tf.Variable(tf.zeros([512], dtype=tf.float32)) 
w2 = tf.Variable(tf.zeros([512, 10], dtype=tf.float32)) 
b2 = tf.Variable(tf.zeros([10], dtype=tf.float32)) 

h = tf.nn.relu(tf.matmul(x, w1) + b1) 
y = tf.matmul(h, w2) + b2 

cross_entropy = tf.nn.sigmoid_cross_entropy_with_logits(labels=y_, logits=y) 
train_step = tf.train.AdamOptimizer().minimize(cross_entropy) 

with tf.Session() as sess: 

    sess.run(tf.initialize_all_variables()) 
    start = time.time() 

    for i in range(20000): 
     batch = mnist.train.next_batch(50) 
     train_step.run(feed_dict={x: batch[0], y_: batch[1]}) 

しかし、次のエラーが繰り返されていますが、これはテンソルフローのドキュメントと矛盾しているようです。

Traceback (most recent call last): 
File "mnist_test.py", line 19, in <module> 
cross_entropy = tf.nn.sigmoid_cross_entropy_with_logits(labels=y_, logits=y) 
TypeError: sigmoid_cross_entropy_with_logits() got an unexpected keyword argument 'labels' 

助けてください!

答えて

3

TensorFlow 1.0.0以上のキーワード引数labelsonly exists。私はあなたが0.12以下を使用していると思います。確認するにはpip freezeまたはprint('TensorFlow version: {0}'.format(tf.__version__))を使用してください。


以前のバージョンのドキュメントは、あなたが使用することができ、以前のバージョンのドキュメントのいくつかの情報を検索するにはhttps://www.tensorflow.org/versions/

で見つけることができます:https://www.google.com/search?q=site:https://www.tensorflow.org/versions/r0.12+sigmoid_cross_entropy_with_logits()

+0

誰かが古いテンソルへのリンクを提供することができますフロードキュメント?私は彼らのウェブサイトでそれを見つけることができません。 –

+0

@JoshuaHowardが答えに追加されました –

関連する問題