def plot_loss(history: keras.callbacks.History, xlabel: str = None, ylabel: str = None, ylim: list = None):
plt.plot(history.history['loss'], label='loss')
plt.plot(history.history['val_loss'], label='val_loss')
plt.ylim(ylim)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.legend()
plt.grid(True)
Last active
March 16, 2022 00:16
-
-
Save gpolyn/3e79d02ea222520757e108e149d8e970 to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment