Tensorflow 多标号张量流的预测

如何在tensorflow图中获得预测向量

predict=tf.argmax（y）

？（因为argmax仅适用于softmax分类器）

我有一个多标签分类问题，因此需要：

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

希望这有助于：

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

希望这有助于：

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