Tensorflow преобразовывает предсказанные значения в двоичные

Question

Я создал нейронную сеть, которая должна классифицировать человека как составляющую более 50 тыс. Или менее. Когда я выдаю предсказание, я получаю такие значения, как [2.06434059 -2.0643425]. Но мне нужно, чтобы они были в [1, 0] или [0, 1]. Есть ли функция тензорного потока, которая будет конвертировать предсказания, или мне нужно сделать это вручную?

Заранее благодарен

Answer 1

Возьмите softmax вашего выхода. например.

output = tf.nn.softmax(input) 

Он преобразует значения в распределение вероятности, где значения будут суммироваться с одним.

В противном случае, если вы просто хотите, чтобы значение {1,0} принимало максимум значений.