def load_test_data(self, data_dim=2, one_hot=True, norm=True):
print('Extract test data: [mnist] from {}'.format(self.data_path))
data = sio.loadmat('{}/mnist_test.mat'.format(self.data_path))
images = data['images'].astype(np.float32)
labels = data['labels'].astype(np.float32)
images, labels = self.preprocess(images,
labels,
data_dim=data_dim,
one_hot=one_hot,
norm=norm)
# package into iterator
assert self.package == None
self.package = iter.Iterator(images, labels)
def load_train_data(self, data_dim=4, one_hot=True, value_norm=True):
print('Extract train data: [cifar10] from {}'.format(self.data_path))
data = sio.loadmat('{}/cifar10_train.mat'.format(self.data_path))
images = data['images'].astype(np.float32)
labels = data['labels'].astype(np.float32)
images, labels = self.preprocess(images,
labels,
data_dim=data_dim,
one_hot=one_hot,
value_norm=value_norm)
# package into iterator
assert self.package == None
self.package = iter.Iterator(images, labels)