PV-2007: Update scatter plots function to use correct scores file bas…

…ed on oos parameter.
GeoscienceAustralia · May 6, 2024 · 34aa5db · 34aa5db
1 parent 2e72d02
commit 34aa5db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/uncoverml/validate.py b/uncoverml/validate.py
@@ -647,7 +647,8 @@ def plot_feature_correlation_matrix(config: Config, x_all, oos_val):
 
 
 def validation_scatter(config: Config, y_true, predictions, oos_val):
-    scores_file = os.path.join(config.output_dir, config.name + "_validation_scores.json")
+    file_name_suffix = '_oos' if oos_val else ''
+    scores_file = os.path.join(config.output_dir, config.name + f"_validation_scores{file_name_suffix}.json")
     with open(scores_file, 'r') as f:
         scores = json.load(f)