mirror of
https://github.com/Deepshift/DeepCreamPy.git
synced 2024-11-28 11:09:42 +00:00
15 lines
315 B
Python
15 lines
315 B
Python
import os
|
|
import numpy as np
|
|
|
|
def load(dir_='./training_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)
|
|
|