import matplotlib.pyplot as plt
def plot_training(history):
plt.figure(figsize=(8, 5)) # 8x5 inches
for key in history.keys():
plt.plot(history[key], label=key)
plt.grid(True)
plt.ylim(0, 1)
plt.xlabel('Epoch')
plt.ylabel('Value')
plt.title('Learning Curve')
plt.legend()
plt.show()
training_history = model.fit(...)
plot_history(training_history)
Example:
Top comments (0)