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