From 3a112da3bb1dafa97800d89396251249a60e06e7 Mon Sep 17 00:00:00 2001
From: kohlhaasrebecca <rebecca.kohlhaas@outlook.com>
Date: Tue, 23 Apr 2024 17:12:13 +0200
Subject: [PATCH] Add early stop in inference for model comparison
---
src/bayesvalidrox/bayes_inference/bayes_inference.py | 3 +++
src/bayesvalidrox/bayes_inference/bayes_model_comparison.py | 1 +
2 files changed, 4 insertions(+)
diff --git a/src/bayesvalidrox/bayes_inference/bayes_inference.py b/src/bayesvalidrox/bayes_inference/bayes_inference.py
index 1898a8ae6..00971b92a 100644
--- a/src/bayesvalidrox/bayes_inference/bayes_inference.py
+++ b/src/bayesvalidrox/bayes_inference/bayes_inference.py
@@ -446,6 +446,9 @@ class BayesInference:
# BME = BME + BMECorrFactor
if self.emulator:
self.log_BME += self.log_BME_corr_factor
+
+ if self.bmc:
+ return self
# ---------------- Parameter Bayesian inference ----------------
if self.inference_method.lower() == 'mcmc':
diff --git a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
index 828613556..fd01689d7 100644
--- a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
+++ b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
@@ -169,6 +169,7 @@ class BayesModelComparison:
# Pass justifiability data as perturbed data
BayesOpts.perturbed_data = justData
BayesOpts.just_analysis = justifiability
+ BayesOpts.bmc = True
bayes_dict[model] = BayesOpts.create_inference()
print("-"*20)
--
GitLab