in train/train.py [0:0]
def load_training_data(base_dir):
x_train = np.load(os.path.join(base_dir, 'train_data.pkl'),allow_pickle=True)
y_train = np.load(os.path.join(base_dir, 'train_labels.pkl'),allow_pickle=True)
return x_train, y_train