Keras输出层阶跃treshold

我正在尝试建立一个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。我知道我们不能这样做，因为随机梯度下降。但是，有没有任何步骤层，我可以设置一个treshold，使我得到0或1作为输出？所需输出的形状为（1,25）

对于二进制输出，应在输出处使用

sigmoid

激活和

binary\u交叉熵

损失。然后，您可以简单地对输出进行四舍五入或应用阈值0.5，以获得二进制0/1值