From a143123abc31eacdd3574a2a77dc18d13f2eeddc Mon Sep 17 00:00:00 2001
From: kohlhaasrebecca <rebecca.kohlhaas@outlook.com>
Date: Wed, 9 Oct 2024 12:25:24 +0200
Subject: [PATCH] [fix] Add output-option to check_reg_quality
Also added related PCE-tests
---
src/bayesvalidrox/post_processing/post_processing.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/src/bayesvalidrox/post_processing/post_processing.py b/src/bayesvalidrox/post_processing/post_processing.py
index f87d00096..e09904f48 100644
--- a/src/bayesvalidrox/post_processing/post_processing.py
+++ b/src/bayesvalidrox/post_processing/post_processing.py
@@ -849,7 +849,11 @@ class PostProcessing:
return self.total_sobol
# -------------------------------------------------------------------------
+<<<<<<< HEAD
def check_reg_quality(self, n_samples: int = 1000, samples=None, outputs: dict = None) -> None:
+=======
+ def check_reg_quality(self, n_samples=1000, samples=None, outputs=None):
+>>>>>>> 4281a0b3 ([fix] Add output-option to check_reg_quality)
"""
Checks the quality of the metamodel for single output models based on:
https://towardsdatascience.com/how-do-you-check-the-quality-of-your-regression-model-in-python-fa61759ff685
@@ -877,7 +881,11 @@ class PostProcessing:
# Evaluate the original and the surrogate model
if outputs is None:
y_val = self._eval_model(samples, key_str='valid')
+<<<<<<< HEAD
else:
+=======
+ else:
+>>>>>>> 4281a0b3 ([fix] Add output-option to check_reg_quality)
y_val = outputs
y_pce_val, _ = self.engine.eval_metamodel(samples=samples)
--
GitLab