Tensorflow earlying_stop,dropout
程序员文章站
2024-03-15 11:31:17
...
都是为了防止过拟合
tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(512, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')
])
上一篇： 损失函数对模型训练结果的影响
下一篇： Tensorboard“无法显示此网站”