From 0b80af2fa46d03d804f3e6871fc38965e879728b Mon Sep 17 00:00:00 2001 From: Farid Mohammadi <farid.mohammadi@iws.uni-stuttgart.de> Date: Fri, 3 Jun 2022 12:12:48 +0200 Subject: [PATCH] [bayes_inference][bayes_model_comparison] fix small bugs. --- .gitignore | 2 ++ src/bayesvalidrox/bayes_inference/bayes_inference.py | 1 + src/bayesvalidrox/bayes_inference/bayes_model_comparison.py | 4 +++- 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index 61449c637..423a52d16 100644 --- a/.gitignore +++ b/.gitignore @@ -4,11 +4,13 @@ *.png !docs/logo/*.png *.svg +!examples/model-comparison/reference/*.svg *.zip *.vtu *.vtp *.pvd *.pkl +!examples/model-comparison/reference/*.pkl # Ignore __pycache__ directories */__pycache__/ diff --git a/src/bayesvalidrox/bayes_inference/bayes_inference.py b/src/bayesvalidrox/bayes_inference/bayes_inference.py index cb7d871ec..714d10a60 100644 --- a/src/bayesvalidrox/bayes_inference/bayes_inference.py +++ b/src/bayesvalidrox/bayes_inference/bayes_inference.py @@ -524,6 +524,7 @@ class BayesInference: for xi in range(yi): ax = axes[yi, xi] ax.set_xlim(MetaModel.bound_tuples[xi]) + plt.close() # Turn off gridlines for ax in figPosterior.axes: diff --git a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py index 7ce17841e..e054d23ef 100644 --- a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py +++ b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py @@ -36,6 +36,7 @@ class BayesModelComparison: # Compute total number of the measurement points MetaModel = list(modelDict.items())[0][1] + MetaModel.ModelObj.read_observation() self.n_meas = MetaModel.ModelObj.n_obs # Generate justifiability data @@ -168,9 +169,10 @@ class BayesModelComparison: ms=10, linewidth=2, label=model ) - plt.title(name.replace('_', '$-$')) + plt.title(f"Data generated by: {name.replace('_', '$-$')}") plt.ylabel("Weights") plt.xlabel("No. of measurement points") + ax.set_xticks(list(range(1, self.n_meas+1))) plt.legend(loc="best") fig.savefig( f'{directory}modelWeights_{name}.svg', bbox_inches='tight' -- GitLab