首页 > 其他 > 详细

TensorFlow keras dropout层

时间:2020-02-03 19:57:33      阅读:190      评论:0      收藏:0      [点我收藏+]
# 建立神经网络模型
    model = keras.Sequential([
        keras.layers.Flatten(input_shape=(28, 28)),  # 将输入数据的形状进行修改成神经网络要求的数据形状
        keras.layers.Dense(128, activation=tf.nn.relu),  # 定义隐藏层,128个神经元的网络层
        keras.layers.Dropout(0.2),
        keras.layers.Dense(10, activation=tf.nn.softmax)  # 10个类别的分类问题,输出神经元个数必须跟总类别数量相同
    ])

  

TensorFlow keras dropout层

原文:https://www.cnblogs.com/LiuXinyu12378/p/12256802.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!