Keras выходной слой шаг порога

Question

Я пытаюсь построить модель Keras.

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(),
  tf.keras.layers.Dropout(0.2),  
  tf.keras.layers.Dense(50, activation='sigmoid'),
  tf.keras.layers.Dense(50, activation='linear'),
  tf.keras.layers.Dense(100, activation='sigmoid'),
  tf.keras.layers.Dense(25, activation='softmax'),
  tf.keras.layers.ReLU(max_value=1)
])

Хотя желаемый результат должен быть 1 или 0. Я понимаю, что мы не можем этого сделать из-за стохастического градиентного спуска. Но есть ли ступенчатый слой, в котором я могу установить порог, чтобы я получал 0 или 1 в качестве результата? Желаемый результат имеет форму (1,25)

Answer 1

Для двоичного выхода следует использовать активацию sigmoid на выходе и потерю binary_crossentropy. Затем вы можете просто округлить или применить порог 0,5 к выходу, чтобы получить двоичные значения 0/1.