def get_train_data()

in labs/04_workflow_sm_pipelines/code/train.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def get_train_data(train_dir):

    x_train = np.load(os.path.join(train_dir, 'x_train.npy'))
    y_train = np.load(os.path.join(train_dir, 'y_train.npy'))
    print('x train', x_train.shape,'y train', y_train.shape)

    return x_train, y_train