def testMC(self):
"""Checks true samples from the model against the ground truth."""
model = neals_funnel.NealsFunnel(ndims=3)
self.validate_ground_truth_using_monte_carlo(
model,
num_samples=int(1e6),
)
def testBasic(self):
"""Checks that you get finite values given unconstrained samples.
We check `unnormalized_log_prob` as well as the values of the sample
transformations.
"""
model = neals_funnel.NealsFunnel(ndims=3)
self.validate_log_prob_and_transforms(
model,
sample_transformation_shapes=dict(identity=[3]),
check_ground_truth_mean=True,
check_ground_truth_standard_deviation=True,
)