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