首页 > 其他 > 详细

神经网络梯度下降的三种学习方式

时间:2016-04-05 07:06:05      阅读:185      评论:0      收藏:0      [点我收藏+]

 

# Batch gradient descent(批量梯度下降)
for i in range(nb_epochs):
    params_grad = evaluate_gradient(loss_function, data, params)
    params = params - learning_rate * params_grad


# Stochastic gradient descent(随机梯度下降)
for i in range(nb_epochs):
    np.random.shuffle(data)
    for example in data:
        params_grad = evaluate_gradient(loss_function, example, params)
        params = params - learning_rate * params_grad


# Mini-batch gradient descent(小批量梯度下降)
for i in range(nb_epochs):
    np.random.shuffle(data)
    for batch in get_batches(data, batch_size=50):
        params_grad = evaluate_gradient(loss_function, batch, params)
        params = params - learning_rate * params_grad

 

神经网络梯度下降的三种学习方式

原文:http://www.cnblogs.com/hhh5460/p/5353632.html

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