Skip to content

Instantly share code, notes, and snippets.

@sid321axn

sid321axn/ANNcompile

Created August 26, 2020 12: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/8d9c71b8b67b363bc1f315f1298dfb0c to your computer and use it in GitHub Desktop.
Save sid321axn/8d9c71b8b67b363bc1f315f1298dfb0c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ANNcompile
model.compile(loss=tf.keras.metrics.mean_squared_error, metrics=[tf.keras.metrics.RootMeanSquaredError(name='rmse')], optimizer='adam')
early_stop = EarlyStopping(monitor='loss', patience=2, verbose=1)
history = model.fit(X_train, y_train, epochs=200, batch_size=1, verbose=1, callbacks=[early_stop], shuffle=False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment