Tensorflow将预测值转换为二进制

Question

我创建了一个神经网络，应该将一个人归类为超过50k或更少。当我输出预测时，我得到像[2.06434059 -2.0643425]这样的值。但我需要它们在[1,0]或[0,1]中。是否有任何张量流函数可以转换预测，还是我必须手动完成？

提前致谢

Answer 1

取输出的softmax。例如

output = tf.nn.softmax(input)

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

否则，如果你只想要{1,0}，只需取出值的最大值。