InfiniteMusic/dataInit.py

14 lines
227 B
Python

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)