SHARE
TWEET

Untitled

a guest Jul 22nd, 2019 67 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. predictions = {
  2.         "classes": tf.argmax(input=logits, axis=1),
  3.         "probabilities": tf.nn.softmax(logits, name="softmax_tensor")
  4.     }
  5.     if mode == tf.estimator.ModeKeys.PREDICT:
  6.       return tf.estimator.EstimatorSpec(mode=mode, predictions=predictions)
  7.      
  8. predictions = {
  9.     "classes": tf.argmax(input=logits, axis=1),
  10.     "probabilities": tf.nn.softmax(logits, name="softmax_tensor")
  11. }
  12. if mode == tf.estimator.ModeKeys.PREDICT:
  13.   return tf.estimator.EstimatorSpec(mode=mode, predictions=predictions)
RAW Paste Data
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy. OK, I Understand
 
Top