Тензор потока всегда предсказывает 0

Question

Я пытаюсь обучить NN для бинарной классификации, но использую подход нескольких классов, следовательно, Class (C) =2. Точность как для обучающих, так и для тестовых наборов данных составляет ~ 96%. Однако ручная проверка показывает, что Tensorflow всегда выбирает 0 для каждого примера.

Мне интересно знать, как отладить эту проблему и что я делаю неправильно. Очень ценю вашу помощь. Благодарю.

(Пожалуйста, дайте мне знать, если я слишком расплывчен и не предоставляю достаточно информации, чтобы вы могли найти правильное решение. Спасибо.)

Answer 1

Если у вас точность 96%, а все прогнозы равны 0, ваш набор данных, вероятно, несбалансирован. Вы должны сбалансировать это и сделать количество образцов из положительных и отрицательных классов одинаковыми. Вы также можете добавить веса для класса, штрафуя больше класса с меньшим количеством образцов.