def testCreateDataset(self):
"""Checks that creating a dataset works."""
# Technically this is private functionality, but we don't have it tested
# elsewhere.
model = log_gaussian_cox_process.LogGaussianCoxProcess(
**_test_dataset())
model2 = log_gaussian_cox_process.LogGaussianCoxProcess(
**model._sample_dataset(tfp_test_util.test_seed()))
self.validate_log_prob_and_transforms(
model2,
sample_transformation_shapes=dict(identity={
'amplitude': [],
'length_scale': [],
'log_intensity': [10],
}, ))
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 = log_gaussian_cox_process.LogGaussianCoxProcess(
**_test_dataset())
self.validate_log_prob_and_transforms(
model,
sample_transformation_shapes=dict(identity={
'amplitude': [],
'length_scale': [],
'log_intensity': [10],
}, ))