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

15
cnn.py
View File

@ -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()
# 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()