i

Ваш вопрос вырван из контекста и без коренной формулировки проблемы получается общим.
Важно понимать, что даже правильно построенная и обученная сеть даст вероятность: input: [0, 1], output: [0.11] и input: [1, 0], output: [0.99], а после применения коэффициента порога доверия к ответу сети получаем ожидаемый результат, например:

parsePrediction(prediction, threshold = 0.5)  {
  if (prediction > threshold) {
    return 1
  } else {
    return 0
  }
}

Посмотрите эту реализацию, коих действительно много, попробуйте слетать альтернативную реализацию и понимание придет.