Python EightSchools Examples

Example #1

File: eight_schools_test.py Project: zhengzhuang3/probability

 def testEightSchools(self):
     """Checks that unconstrained parameters yield finite joint densities."""
     model = eight_schools.EightSchools()
     self.validate_log_prob_and_transforms(
         model,
         sample_transformation_shapes=dict(identity={
             'avg_effect': [],
             'log_stddev': [],
             'school_effects': [8],
         }))

Example #2

File: eight_schools_test.py Project: mederrata/probability

  def testEightSchoolsHMC(self):
    """Checks approximate samples from the model against the ground truth."""
    model = eight_schools.EightSchools()

    self.validate_ground_truth_using_hmc(
        model,
        num_chains=4,
        num_steps=4000,
        num_leapfrog_steps=10,
        step_size=0.4,
    )