tf_prediction = tf.nn.softmax(tf_output, axis=1).numpy().argmax(axis=-1)

Hi. Softmax returns array with two numbers.
9
1
Andrej Baranovskij
Yashu Gupta
·Follow
Jan 11, 2021
--
tf_prediction = tf.nn.softmax(tf_output, axis=1).numpy().argmax(axis=-1)
--
--
Written by Yashu Gupta80 Followers
·47 Following
Lead Data Scientist |AI Researcher | NLP Evangelist
Responses (1)
Help
Status
About
Careers
Press
Blog
Privacy
Terms
Text to speech
Teams