def setUp(self):
self.priors = prior.PriorDict(
dict(
phase=prior.Uniform(minimum=0.0, maximum=2 * np.pi),
psi=prior.Cosine(minimum=0.0, maximum=np.pi),
))
self.jump_proposal = proposal.DrawFlatPrior(priors=self.priors)
name='geocent_time', latex_label='$t_c$', unit='$s$')
for key in ['a_1', 'a_2', 'tilt_1', 'tilt_2', 'phi_12', 'phi_jl', 'geocent_time']:
priors[key] = injection_parameters[key]
likelihood = bilby.gw.GravitationalWaveTransient(
interferometers=ifos, waveform_generator=waveform_generator)
# Definition of the custom jump proposals. Define a JumpProposalCycle. The first argument is a list
# of all allowed jump proposals. The second argument is a list of weights for the respective jump
# proposal.
jump_proposals = proposal.JumpProposalCycle(
[proposal.EnsembleWalk(priors=priors),
proposal.EnsembleStretch(priors=priors),
proposal.DifferentialEvolution(priors=priors),
proposal.EnsembleEigenVector(priors=priors),
bilby.gw.sampler.proposal.SkyLocationWanderJump(priors=priors),
bilby.gw.sampler.proposal.CorrelatedPolarisationPhaseJump(priors=priors),
bilby.gw.sampler.proposal.PolarisationPhaseJump(priors=priors),
proposal.DrawFlatPrior(priors=priors)],
weights=[2, 2, 5, 1, 1, 1, 1, 1])
# Run cpnest with the proposals kwarg specified.
# Make sure to have a version of cpnest installed that supports custom proposals.
result = bilby.run_sampler(
likelihood=likelihood, priors=priors, sampler='cpnest', npoints=1000,
injection_parameters=injection_parameters, outdir=outdir, label=label,
proposals=jump_proposals)
# Make a corner plot.
result.plot_corner()