From a31e50d5ea20cea69517dc231c225d7646b7a438 Mon Sep 17 00:00:00 2001
From: Farid Mohammadi <farid.mohammadi@iws.uni-stuttgart.de>
Date: Tue, 31 May 2022 14:11:33 +0200
Subject: [PATCH] [surrogate][sequential] wrap kdeplot for the likelihood plot
in a try and except block.
---
.../surrogate_models/sequential_design.py | 13 ++++++++-----
1 file changed, 8 insertions(+), 5 deletions(-)
diff --git a/src/bayesvalidrox/surrogate_models/sequential_design.py b/src/bayesvalidrox/surrogate_models/sequential_design.py
index f1b765b76..a179bf699 100644
--- a/src/bayesvalidrox/surrogate_models/sequential_design.py
+++ b/src/bayesvalidrox/surrogate_models/sequential_design.py
@@ -217,7 +217,7 @@ class SeqDesign():
Ynew, _ = Model.run_model_parallel(
Xnew, prevRun_No=total_n_samples
)
-
+ total_n_samples += Xnew.shape[0]
# ------ Plot the surrogate model vs Origninal Model ------
if hasattr(PCEModel, 'adapt_verbose') and \
PCEModel.adapt_verbose:
@@ -1869,10 +1869,13 @@ class SeqDesign():
idx = len([name for name in os.listdir(newpath) if 'Likelihoods_'
in name and os.path.isfile(os.path.join(newpath, name))])
fig, ax = plt.subplots()
- sns.kdeplot(np.log(valid_likelihoods[valid_likelihoods > 0]),
- shade=True, color="g", label='Ref. Likelihood')
- sns.kdeplot(np.log(Likelihoods[Likelihoods > 0]), shade=True,
- color="b", label='Likelihood with PCE')
+ try:
+ sns.kdeplot(np.log(valid_likelihoods[valid_likelihoods > 0]),
+ shade=True, color="g", label='Ref. Likelihood')
+ sns.kdeplot(np.log(Likelihoods[Likelihoods > 0]), shade=True,
+ color="b", label='Likelihood with PCE')
+ except:
+ pass
text = f"Hellinger Dist.={distHellinger:.3f}\n logBME={logBME:.3f}"
"\n DKL={KLD:.3f}"
--
GitLab