Birden çok etiket için tensör akışında tahmin

Question

Tensorlow grafiğinde bir tahmin vektörünü predict = tf.argmax(y) dan nasıl alabilirim?

ben gibi bir şey gerekiyor bu nedenle bir çok etiketli sınıflandırma sorun var (çünkü argmax sadece SoftMax sınıflandırıcılar için çalışır):

predictions = [1. if prob > 0.5 else 0. for prob in y] 

Answer 1

Umut bu yardımcı olur:

import tensorflow as tf 
import numpy as np 
sess = tf.InteractiveSession() 
prob = tf.constant(np.random.rand(10)) 
predictions = tf.select(prob > 0.5, tf.ones_like(prob), tf.zeros_like(prob)) 
print(predictions.eval())