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]))
