From 12e638cec3d1dc784119f65d43ca56d99db8cc41 Mon Sep 17 00:00:00 2001 From: klara Date: Sun, 26 Jan 2025 14:14:55 +0100 Subject: [PATCH] color change --- cnn.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/cnn.py b/cnn.py index 41f38e2..8272d3a 100644 --- a/cnn.py +++ b/cnn.py @@ -195,10 +195,11 @@ if __name__ == "__main__": print(f"Final Accuracy: {accuracy:.4f}") # Visualize Losses - plt.plot(train_losses, label="Train Loss") - plt.plot(val_losses, label="Validation Loss") - plt.xlabel("Epochs") - plt.ylabel("Loss") - plt.title("Loss Curve") - plt.legend() - plt.show() \ No newline at end of file + # Visualize Losses with custom colors +plt.plot(train_losses, label="Train Loss", color='#0032ff') # Blue color for train loss +plt.plot(val_losses, label="Validation Loss", color='#0fff00') # Green color for validation loss +plt.xlabel("Epochs") +plt.ylabel("Loss") +plt.title("Loss Curve") +plt.legend() +plt.show() \ No newline at end of file