Generate training data for spectrogram

spectrogram_build_training_data.py

def build_training_data(batch_size, sample_size):
  xs = np.random.randn(batch_size, sample_size)
  f, t, ys = spectrogram(xs)

  (num_rows, num_cols) = (ys.shape[1], ys.shape[2])
  print(ys.shape)

  ys = ys.reshape(batch_size, num_rows * num_cols)
  Ys = np.hstack([ys.real, ys.imag])
  return (xs, Ys, num_rows, num_cols)