precompute training data outside training script
This commit is contained in:
13
dataInit.py
Normal file
13
dataInit.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import data
|
||||
import numpy as np
|
||||
|
||||
x,y = data.dataset(data.load())
|
||||
size=len(x)
|
||||
|
||||
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)
|
||||
Reference in New Issue
Block a user