Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Keras model for training spectrogram NN
import tensorflow as tf
size = 2048
N = 15000
(xs, ys, rows, cols) = build_training_data(N, size)
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(rows * cols * 2, input_dim=size, use_bias=False)
])
model.summary()
optimizer = tf.keras.optimizers.Adam(amsgrad=True)
model.compile(loss='mean_squared_error', optimizer=optimizer)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment