def __iter__(self):
x_train, y_train, label_train, x_test, y_test, label_test = mnist_dataset.read_data()
batch_nums = math.ceil(len(x_test) / self.batch_size)
for i in range(batch_nums - 1):
begin = i * self.batch_size
end = (i + 1) * self.batch_size
yield x_test[begin: end], label_test[begin: end]
begin = (batch_nums - 1) * self.batch_size
yield x_test[begin:], label_test[begin:]
def val_data():
x_train, y_train, label_train, x_test, y_test, label_test = mnist_dataset.read_data(
)
return x_test, y_test, label_test