tensorflow2.0 加入dropout层
时间: 2023-10-28 11:05:34 浏览: 214
在 TensorFlow 2.0 中,可以使用 `tf.keras.layers.Dropout` 层添加 dropout。该层的参数是一个浮点数,表示要将输入单元随机设置为 0 的比率。
示例代码如下:
```
import tensorflow as tf
model = tf.keras.Sequential([
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')
])
```
上面的代码中,使用了一个 `Dense` 层,一个 `Dropout` 层和一个输出层。`Dropout` 层的参数是 0.2,表示将输入单元的 20% 随机设置为 0。
阅读全文