import data
import numpy as np

x,y = data.dataset(data.load())
size=len(x)
print(size)

x_np = np.stack(x)
x_np = np.expand_dims(x_np, axis=1)

y_np = np.stack(y)
y_np = np.expand_dims(y_np, axis=1)

np.savez_compressed("data",x_np,y_np)