舍弃—Dropout

作用:为了缓解神经网络过拟合

方法:在神经网络训练时,将一部分神经元按照一定的概率从神经网络中暂时舍弃。神经网络使用时,被舍弃的神经元恢复链接。

  

TF2描述:

  tf.keras.layers.Dropout(舍弃的概率)

实例: 

  

1 model = tf.keras.models.Sequential(
2     Conv2D(filter=6, kernel_size=(5, 5), padding='same'),  # 卷积层
3     BatchNormalization(),  # BN层
4     Activation('relu'),  # 激活函数
5     MaxPool2D(pool_size=(2, 2), strides=2, padding='same'),  # 池化层
6     Dropout(0.2)  # dropout层
7 )

猜你喜欢

转载自www.cnblogs.com/cropmodel/p/13365238.html