tensorflow学习之三 (添加批规范化减少过拟合,提高准确率)

65 篇文章 458 订阅 ¥9.90 ¥99.00
33 篇文章 7 订阅 ¥29.90 ¥99.00

在这我以mnist数据集为例,简单的搭建了一个7层网络,其余参数与前一博客相同,此代码结合前一篇即可完成自己的数据集训练。

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True, reshape=False)

def fully_connected(prev_layer, num_units, is_training):
    #fc_layer=tf.layers.Dense(prev_layer,num_units)
    fc_layer=tf.layers.dense(prev_layer,num_units)
    fc_layer_Bn=tf.layers.batch_normalization( fc_layer,training=is_training)
    layer=tf.nn.relu(fc_layer_Bn)
    return layer
def conv_layer(pre_layer,layer_depth,is_training):
    strides = 2 if layer_depth % 3 == 0 else 1
    conv_layer_cov=tf.layers.conv2d(pre_layer,layer_depth*4,3,strides,"same",use_bias=None,activation=None)
    conv_layer_cov=tf.layers.batch_normalization(conv_layer_cov,training=is_training)
    conv_layer_cov=tf.nn.relu(conv_layer_cov)
    return conv_la
  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xiao__run

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值