DeepCreamPy/src/load.py
2018-02-09 17:52:53 -05:00

15 lines
307 B
Python

import os
import numpy as np
def load(dir_='../data/npy'):
x_train = np.load(os.path.join(dir_, 'x_train.npy'))
x_test = np.load(os.path.join(dir_, 'x_test.npy'))
return x_train, x_test
if __name__ == '__main__':
x_train, x_test = load()
print(x_train.shape)
print(x_test.shape)