python - Tensorflow 将预测值转换为二进制

Question

我创建了一个神经网络，该网络应该将一个人分类为收入超过 5 万或更少。当我输出预测时，我得到的值类似于 [2.06434059 -2.0643425]。但我需要它们位于 [1, 0] 或 [0, 1] 中。是否有任何 tensorflow 函数可以转换预测，或者我必须手动执行此操作？

提前致谢

Answer 1

获取输出的 softmax。例如。

output = tf.nn.softmax(input)

它将这些值转换为概率分布，其中这些值的总和为 1。

否则，如果您只想要 {1,0}，只需取传入值的最大值即可。