Revision 2c9966337248b210bd32e36cff3afb19ef0f1c09 authored by James Hensman on 24 October 2016, 12:24:29 UTC, committed by GitHub on 24 October 2016, 12:24:29 UTC
1 parent 35b07ef
test_transforms.py
import GPflow
import tensorflow as tf
import numpy as np
import unittest
from GPflow import settings
float_type = settings.dtypes.float_type
np_float_type = np.float32 if float_type is tf.float32 else np.float64
class TransformTests(unittest.TestCase):
def setUp(self):
tf.reset_default_graph()
self.x = tf.placeholder(float_type)
self.x_np = np.random.randn(10).astype(np_float_type)
self.session = tf.Session()
self.transforms = [C() for C in GPflow.transforms.Transform.__subclasses__()]
self.transforms.append(GPflow.transforms.Logistic(7.3, 19.4))
def test_tf_np_forward(self):
"""
Make sure the np forward transforms are the same as the tensorflow ones
"""
ys = [t.tf_forward(self.x) for t in self.transforms]
ys_tf = [self.session.run(y, feed_dict={self.x: self.x_np}) for y in ys]
ys_np = [t.forward(self.x_np) for t in self.transforms]
for y1, y2 in zip(ys_tf, ys_np):
self.assertTrue(np.allclose(y1, y2))
def test_forward_backward(self):
ys_np = [t.forward(self.x_np) for t in self.transforms]
xs_np = [t.backward(y) for t, y in zip(self.transforms, ys_np)]
for x in xs_np:
self.assertTrue(np.allclose(x, self.x_np))
def test_logjac(self):
"""
We have hand-crafted the log-jacobians for speed. Check they're correct
wrt a tensorflow derived version
"""
# there is no jacobian: loop manually
def jacobian(f):
return tf.pack([tf.gradients(f(self.x)[i], self.x)[0] for i in range(10)])
tf_jacs = [tf.log(tf.matrix_determinant(jacobian(t.tf_forward))) for t in self.transforms if
type(t) is not GPflow.transforms.LowerTriangular]
hand_jacs = [t.tf_log_jacobian(self.x) for t in self.transforms if
type(t) is not GPflow.transforms.LowerTriangular]
for j1, j2 in zip(tf_jacs, hand_jacs):
self.assertTrue(np.allclose(self.session.run(j1, feed_dict={self.x: self.x_np}),
self.session.run(j2, feed_dict={self.x: self.x_np})))
class TestLowerTriTransform(unittest.TestCase):
"""
Some extra tests for the LowerTriangle transformation.
"""
def setUp(self):
self.t = GPflow.transforms.LowerTriangular(3)
def testErrors(self):
self.t.free_state_size((6, 6, 3))
with self.assertRaises(ValueError):
self.t.free_state_size((6, 6, 2))
with self.assertRaises(ValueError):
self.t.free_state_size((7, 6, 3))
self.t.forward(np.ones(3 * 6))
with self.assertRaises(ValueError):
self.t.forward(np.ones(3 * 7))
if __name__ == "__main__":
unittest.main()
Computing file changes ...