3016498
/
ANLP_WS24_CA2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lines dashed

main
arman 2025-02-15 14:23:10 +01:00
parent db3a3474a2
commit 7500367475
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
transformer_bootstrap_agg.py
View File

@ -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')