Missed a change on the last commit

jpata · Nov 13, 2024 · bba3098 · bba3098
1 parent f563823
commit bba3098
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlpf/model/training.py b/mlpf/model/training.py
@@ -318,7 +318,7 @@ def validation_plots(batch, ypred_raw, ytarget, ypred, tensorboard_writer, comet
         comet_experiment.log_figure(figure_name= f'PU Hist - Epoch {epoch} Batch {itrain}')
         plt.close()
 
-        plt.scatter(x, y, alpha=0.5, color='blue')
+        plt.scatter(pred_pu, targ_pu, alpha=0.5, color='blue')
         plt.xlabel('Predicted PU')
         plt.ylabel('Target PU')
         plt.title(f'Scatter Plot of Predicted vs Target PU (Epoch {epoch})')