def load_training_data()

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