Skip to content

Instantly share code, notes, and snippets.

@sid321axn
Created August 26, 2020 14:02
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save sid321axn/d5fd8e9abb045d7dc836e181e5991e4c to your computer and use it in GitHub Desktop.
Save sid321axn/d5fd8e9abb045d7dc836e181e5991e4c to your computer and use it in GitHub Desktop.
model_lstm.compile(loss=tf.keras.metrics.mean_squared_error,
metrics=[tf.keras.metrics.RootMeanSquaredError(name='rmse')], optimizer='adam')
early_stop = EarlyStopping(monitor='loss', patience=5, verbose=1)
history_model_lstm = model_lstm.fit(X_tr_t, y_train, epochs=100, batch_size=1, verbose=1, shuffle=False, callbacks=[early_stop])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment