【学习笔记】# 保存损失函数的值 train_loss_list = [] # 迭代items_num 1000 ...

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

# 保存损失函数的值
train_loss_list = []
# 迭代items_num 10000次
for i in range(items_num):
    # 从数据中抽取出100个数据
    batch_mask = np.random.choice(train_size, batch_size)
    x_batch = x_train[batch_mask]
    # 对应的把这100个样本的监督标签页提取出来
    y_batch = y_train[batch_mask]
    # 利用模型计算梯度
    # 梯度计算方法速度非常非常的慢
#     grad = neural_networks.numerical_gradient(x_batch , y_batch)
    # 反向传播: 相当于一个快速梯度计算方法
    grad = neural_networks.gradient(x_batch,y_batch)
    # 有了梯度矩阵, 根据梯度对所有的参数进行更新
    for key in [\'W1\',\'W2\',\'b1\',\'b2\']:
        neural_networks.params[key] -= learning_rate * grad[key]
    # 记录当前损失函数的值
    loss = neural_networks.loss(x_batch ,y_batch)
   
    # 在迭代的过程中,观察损失函数的变化
    train_loss_list.append(loss)
    # 每迭代100次,我们打印一次损失函数
    if i % 100 == 0 :
        print(\"当前损失函数的值为:\",loss)
   

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：Train items loss rain 学习笔记