Merge pull request #346 from GeoStat-Framework/emcee_vectorize

MuellerSeb · web-flow · commit af19c447bfc2 · 2024-04-19T20:40:06.000+02:00
emcee: vectorize sampling since all ln_pdfs accept numpy arrays
diff --git a/src/gstools/random/rng.py b/src/gstools/random/rng.py
@@ -86,7 +86,7 @@ def sample_ln_pdf(
             self.random.rand(nwalkers).reshape((nwalkers, 1)) * sample_around
         )
         # initialize the sampler
-        sampler = mc.EnsembleSampler(nwalkers, 1, ln_pdf)
+        sampler = mc.EnsembleSampler(nwalkers, 1, ln_pdf, vectorize=True)
         # burn in phase with saving of last position
         initial_state = State(init_guess, copy=True)
         initial_state.random_state = self.random.get_state()

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,7 @@ def sample_ln_pdf(`
`86`	`86`	`self.random.rand(nwalkers).reshape((nwalkers, 1)) * sample_around`
`87`	`87`	`)`
`88`	`88`	`# initialize the sampler`
`89`		`- sampler = mc.EnsembleSampler(nwalkers, 1, ln_pdf)`
	`89`	`+ sampler = mc.EnsembleSampler(nwalkers, 1, ln_pdf, vectorize=True)`
`90`	`90`	`# burn in phase with saving of last position`
`91`	`91`	`initial_state = State(init_guess, copy=True)`
`92`	`92`	`initial_state.random_state = self.random.get_state()`