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