def load_data(self, dataset_path, share = False):
"""Load the data set.
"""
f = gzip.open(dataset_path, 'rb')
train_set, valid_set, test_set = pickle.load(f)
f.close()
# share the data
train_set_x, train_set_y = self.shared_dataset(train_set, train=True)
valid_set_x, valid_set_y = self.shared_dataset(valid_set)
test_set_x, test_set_y = self.shared_dataset(test_set)
if share:
reval = [(train_set_x, train_set_y), (valid_set_x, valid_set_y), (test_set_x, test_set_y)]
else:
reval = [train_set, valid_set, test_set] # NON-shared data (they didn't share the data in the code Crino!!!!!)
return reval
评论列表
文章目录