GuitarLSTM

commit db7afb00ad5056c86b34f2a03300874b4b1e7a80
parent 3d70b8ebff659b8adce726f214c33044c47cc25a
Author: jmiller656 <joshxmiller656@gmail.com>
Date:   Wed,  3 Feb 2021 20:16:07 -0500

Switch to MSE loss

Diffstat:
M
guitar_lstm_colab.ipynb
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/guitar_lstm_colab.ipynb b/guitar_lstm_colab.ipynb
@@ -163,7 +163,7 @@
     "model.add(Conv1D(conv1d_filters, 12,strides=conv1d_strides, activation=None, padding='same'))\n",
     "model.add(LSTM(hidden_units))\n",
     "model.add(Dense(1, activation=None))\n",
-    "model.compile(optimizer=Adam(learning_rate=learning_rate), loss=error_to_signal, metrics=[error_to_signal])\n",
+    "model.compile(optimizer=Adam(learning_rate=learning_rate), loss='mse', metrics=[error_to_signal])\n",
     "model.summary()\n",
     "\n",
     "# Load and Preprocess Data ###########################################\n",