张量流中多个标签的预测

Question

除了 predict = tf.argmax(y) ? （因为 argmax 仅适用于 softmax 分类器）

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

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

张量流中多个标签的预测

prediction in tensor flow for multiple labels

tensorflow