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)