def test_run(self):
model = self.build_model()
with model:
start = {'psi': 0.5, 'z': (self.y > 0).astype('int16'), 'theta': 5}
step_one = pm.Metropolis([model.theta_interval__, model.psi_logodds__])
step_two = pm.BinaryMetropolis([model.z])
pm.sample(50, step=[step_one, step_two], start=start)
def test_run(self):
model = self.build_model()
with model:
start = {
"psi": np.array(0.5, dtype="f"),
"z": (self.y > 0).astype("int16"),
"theta": np.array(5, dtype="f"),
}
step_one = pm.Metropolis([model.theta_interval__, model.psi_logodds__])
step_two = pm.BinaryMetropolis([model.z])
pm.sample(50, step=[step_one, step_two], start=start, chains=1)
