diff options
Diffstat (limited to 'tensorflow/examples/learn/iris_custom_model.py')
-rw-r--r-- | tensorflow/examples/learn/iris_custom_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/learn/iris_custom_model.py b/tensorflow/examples/learn/iris_custom_model.py index 31fb88954b..fbc50716c9 100644 --- a/tensorflow/examples/learn/iris_custom_model.py +++ b/tensorflow/examples/learn/iris_custom_model.py @@ -43,7 +43,7 @@ def my_model(features, target): # Compute logits (1 per class) and compute loss. logits = layers.fully_connected(features, 3, activation_fn=None) - loss = tf.contrib.losses.softmax_cross_entropy(logits, target) + loss = tf.losses.softmax_cross_entropy(target, logits) # Create a tensor for training op. train_op = tf.contrib.layers.optimize_loss( |