def testLorenzSystemHMC(self): """Checks approximate samples from the model against the ground truth.""" model = lorenz_system.LorenzSystem( **_make_dataset(_partially_observed_data)) self.validate_ground_truth_using_hmc( model, num_chains=2, num_steps=2000, num_leapfrog_steps=15, step_size=0.03, )
def testLorenzSystem(self): """Checks that unconstrained parameters yield finite joint densities.""" model = lorenz_system.LorenzSystem(**_test_dataset(), use_markov_chain=True) self.validate_log_prob_and_transforms( model, sample_transformation_shapes=dict(identity=[30, 3]))