From d1a8b85e946f4cce6ce1bdf36c4e2cf33731bb63 Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Fri, 7 Oct 2016 16:34:27 +0200
Subject: [PATCH] Adds ability to set the proposal_scale
---
pyfstat.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pyfstat.py b/pyfstat.py
index bb5fb3c..7d7781c 100755
--- a/pyfstat.py
+++ b/pyfstat.py
@@ -662,7 +662,7 @@ class MCMCSearch(BaseSearchClass):
return p0
- def run(self):
+ def run(self, proposal_scale_factor=None):
if self.old_data_is_okay_to_use is True:
logging.warning('Using saved data from {}'.format(
@@ -679,7 +679,7 @@ class MCMCSearch(BaseSearchClass):
sampler = emcee.PTSampler(
self.ntemps, self.nwalkers, self.ndim, self.logl, self.logp,
logpargs=(self.theta_prior, self.theta_keys, self.search),
- loglargs=(self.search,), betas=self.betas)
+ loglargs=(self.search,), betas=self.betas, a=proposal_scale_factor)
p0 = self.generate_initial_p0()
p0 = self.apply_corrections_to_p0(p0)
--
GitLab