experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bugfix: Make tf.contrib.distributions.Independent tests not flaky.

Browse Source 
PiperOrigin-RevId: 173921378
This commit is contained in:
Joshua V. Dillon 2017-10-30 11:23:36 -07:00 committed by TensorFlower Gardener
parent 4b63f47d9f
commit 629e6d0c10
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/contrib/distributions/python/kernel_tests/independent_test.py
View File

@ -56,7 +56,7 @@ class ProductDistributionTest(test.TestCase):
distribution=normal_lib.Normal(loc=loc, scale=scale), distribution=normal_lib.Normal(loc=loc, scale=scale),
reinterpreted_batch_ndims=1) reinterpreted_batch_ndims=1)
x = ind.sample([4, 5]) x = ind.sample([4, 5], seed=42)
log_prob_x = ind.log_prob(x) log_prob_x = ind.log_prob(x)
x_, actual_log_prob_x = sess.run([x, log_prob_x]) x_, actual_log_prob_x = sess.run([x, log_prob_x])
@ -79,7 +79,7 @@ class ProductDistributionTest(test.TestCase):
scale_identity_multiplier=scale), scale_identity_multiplier=scale),
reinterpreted_batch_ndims=1) reinterpreted_batch_ndims=1)
x = ind.sample([4, 5]) x = ind.sample([4, 5], seed=42)
log_prob_x = ind.log_prob(x) log_prob_x = ind.log_prob(x)
x_, actual_log_prob_x = sess.run([x, log_prob_x]) x_, actual_log_prob_x = sess.run([x, log_prob_x])
@ -141,7 +141,7 @@ class ProductDistributionTest(test.TestCase):
dtypes.float32, shape=logits.shape if static_shape else None) dtypes.float32, shape=logits.shape if static_shape else None)
ind = independent_lib.Independent( ind = independent_lib.Independent(
distribution=bernoulli_lib.Bernoulli(logits=logits_ph)) distribution=bernoulli_lib.Bernoulli(logits=logits_ph))
x = ind.sample(sample_shape) x = ind.sample(sample_shape, seed=42)
log_prob_x = ind.log_prob(x) log_prob_x = ind.log_prob(x)
[ [
x_, x_,