From 7500367475ee41679478034349d20dd4b72c6ad2 Mon Sep 17 00:00:00 2001 From: arman Date: Sat, 15 Feb 2025 14:23:10 +0100 Subject: [PATCH] lines dashed --- transformer_bootstrap_agg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/transformer_bootstrap_agg.py b/transformer_bootstrap_agg.py index 3d35d1d..5cfb764 100644 --- a/transformer_bootstrap_agg.py +++ b/transformer_bootstrap_agg.py @@ -282,7 +282,7 @@ fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(14, 6)) # Plot Train and Validation Losses for i in range(num_models): ax1.plot(range(1, params['epochs'] + 1), all_train_losses[i], label=f"Train Model {i+1}") - ax1.plot(range(1, params['epochs'] + 1), all_val_losses[i], label=f"Val Model {i+1}") + ax1.plot(range(1, params['epochs'] + 1), all_val_losses[i], label=f"Val Model {i+1}", linestyle='dashed') ax1.set_title('Train and Validation Loss') ax1.set_xlabel('Epochs') @@ -292,7 +292,7 @@ ax1.legend() # Plot Train and Validation R² for i in range(num_models): ax2.plot(range(1, params['epochs'] + 1), all_train_r2_scores[i], label=f"Train Model {i+1}") - ax2.plot(range(1, params['epochs'] + 1), all_val_r2_scores[i], label=f"Val Model {i+1}") + ax2.plot(range(1, params['epochs'] + 1), all_val_r2_scores[i], label=f"Val Model {i+1}", linestyle='dashed') ax2.set_title('Train and Validation R²') ax2.set_xlabel('Epochs')