在学习《深度学习入门——基于Python的理论与实现》时,进行mini-batch的实现出现了报错,代码如下
import numpy as np
from dataset.mnist import load_mnist
from two_layer_net import TwoLayerNet
(x_train,t_train),(x_test,t_test)=load_mnist(normalize=True,one_hot_label=True)
train_loss_list=[]
iters_num=10000
train_size=x_train.shape[0]
batch_size=100
learning_rate=0.1
network=TwoLayerNet(input_size=784,hidden_size=50,output_size=10)
for i in range(iters_num):
#获取mini-batch
batch_mask=np.random.choice(train_size,batch_size)
x_batch=x_train[batch_mask]
t_batch=t_train[batch_mask]
#计算梯度
grad=network.numerical_gradient(x_batch,t_batch)
#更新参数
for key in ('W1','b1','W2','b2'):
network.params[key]-=learning_rate*grad[key]
#记录学习过程
loss=network.loss(x_batch,t_batch)
train_loss_list.append(loss)
报错情况如下

我已经把书中的源代码下载进了Anaconda所在的文件夹中,之前调用书中的模块都没有出现报错,现在报错了,不知道是什么原因,友友们可以帮忙看一下吗?谢谢!