【发布时间】:2021-07-17 21:57:02
【问题描述】:
我在密集的 NN 模型中将 dropout 作为正则化方法实施时遇到问题。似乎添加大于 0 的 dropout 值只会缩小预测值,在某种程度上让我认为在将单个权重设置为零后没有正确考虑某些事情。我确定我执行不正确,但我似乎无法弄清楚是什么。
构建此模型的代码直接取自 tensorflow 页面 (https://www.tensorflow.org/tutorials/keras/overfit_and_underfit),但无论我使用什么架构构建模型都会出现。
model = tf.keras.Sequential([
layers.Dense(512, activation='relu', input_shape=[len(X_train[0])]),
layers.Dropout(0.5),
layers.Dense(512, activation='relu'),
layers.Dropout(0.5),
layers.Dense(512, activation='relu'),
layers.Dropout(0.5),
layers.Dense(512, activation='relu'),
layers.Dropout(0.5),
layers.Dense(1)
])
任何帮助将不胜感激!
【问题讨论】:
标签: python tensorflow dropout