color change
parent
8ea3305644
commit
12e638cec3
15
cnn.py
15
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()
|
||||
# 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()
|
||||
Loading…
Reference in New Issue