in tensorflow-train-in-sagemaker-deploy-with-lambda/mnist-2.py [0:0]
def _load_training_data(base_dir):
"""Load MNIST training data"""
x_train = np.load(os.path.join(base_dir, 'train_data.npy'))
y_train = np.load(os.path.join(base_dir, 'train_labels.npy'))
return x_train, y_train