发新帖

model类中想加入dropout解决过拟合,代码应该怎么写

[复制链接]
322 1

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
class Model(tf.Keras.Model):
    def __init__(self):
        super().__init__()
        .............

精彩评论1

祝家鑫  TF荚荚  发表于 6 天前 | 显示全部楼层
本帖最后由 祝家鑫 于 2019-10-17 15:52 编辑

def call(self, input):
    X = keras.layers.Dropout(rate=0.5)(X)
    return X
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

4

积分
快速回复 返回顶部 返回列表