color change

main
klara 2025-01-26 14:14:55 +01:00
parent 8ea3305644
commit 12e638cec3
1 changed files with 8 additions and 7 deletions

5
cnn.py
View File

@ -195,8 +195,9 @@ if __name__ == "__main__":
print(f"Final Accuracy: {accuracy:.4f}") print(f"Final Accuracy: {accuracy:.4f}")
# Visualize Losses # Visualize Losses
plt.plot(train_losses, label="Train Loss") # Visualize Losses with custom colors
plt.plot(val_losses, label="Validation Loss") 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.xlabel("Epochs")
plt.ylabel("Loss") plt.ylabel("Loss")
plt.title("Loss Curve") plt.title("Loss Curve")