複数ラベルのテンソルフロー予測

Question

predict = tf.argmax(y)以外のテンソルフローグラフで予測ベクトルを取得するにはどうすればよいですか？

は私が何か必要なので、したがって、マルチラベル分類問題を持っている（以降ARGMAXのみソフトマックス分類のために働く）：

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

Answer 1

希望をこの役立ちます：

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())