simple vae style model

dataInit.py

@@ -1,14 +1,14 @@
 import data
 import numpy as np
 
-x = data.load()
+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)
+y_np = np.stack(y)
+y_np = np.expand_dims(y_np, axis=1)
 
-np.savez_compressed("data",x_np)
+np.savez_compressed("data",x_np,y_np)