TensorFlow将预测值转换为二进制

Question

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

预先感谢

Answer 1

取出输出的软效果。例如

output = tf.nn.softmax(input)

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

否则，如果您只想{1,0}，只需以最大的含量。