滑动平均模型tf.train.ExponentialMovingAverage(v1, stpe)

滑动平均模型
在采用随机梯度下降算法训练神经网络时候,使用滑动平均模型可以在一定的程度上提高最终的模型在测试数据上的表现

Tensorflow中使用函数提供了下面的函数

>>> tf.train.ExponentialMovingAverage(v1,step)
% v1: variable. step =控制下降的速度
% shadow_variable = decay * shadow_variable + (1 - decay) * variable
% shadow_variable 为影子变量,variable为待更新的变量


% decay = min {decay, (1+step)/(10+step)}:设置动态decay快速更新variable
>>> v1 = tf.Varibele(0, dtype = tf.float32)
>>> step = tf.Variable(0, trainable = False)
>>> ema = tf.train.ExponentialMovingAverage(0.99, step)
>>> maintain_average_op = ema.apply([v1])
>>> with tf.Session() as sess:
        init_op = tf.initialize_all_variable()
        sess.run(init_op)
        print(sess.run([v1, ema.average(v1)])
% 输出:[0.0,  0.0]
        sess.run(tf.assign(v1, 5))
        sess.run(maintain_averages_op)
        print sess.run([v1, ema.average(v1)])
% 输出:[5.0, 4.5]


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值