Keras 丢弃层模型预测

4

Dropout层仅在模型训练期间使用,而不是测试期间使用。

如果我在Keras串行模型中有一个Dropout层,在执行model.predict()之前,我需要做一些事情来删除或禁用它吗?

1个回答

5

不,你不需要将其静音或删除。Keras会自动处理它。

文档中明确提到了这一点。 Keras模型有两种模式:

  1. 训练
  2. 测试

正则化机制,如Dropout和L1 / L2权重正则化,在测试时关闭。

注意:此外,批量归一化是一种比Dropout更受欢迎的正则化技术。建议考虑使用它。


在函数式 API 中是相同的情况吗? - Aaditya Ura

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接