https://github.com/GPflow/GPflow
Revision c0f0fbe79925022e0bae59666c34db78101ddc32 authored by James Hensman on 20 April 2016, 13:40:57 UTC, committed by James Hensman on 20 April 2016, 13:40:57 UTC
Move from fork to main TensorFlow.
Tip revision: c0f0fbe79925022e0bae59666c34db78101ddc32 authored by James Hensman on 20 April 2016, 13:40:57 UTC
Merge pull request #50 from GPflow/tf_main
Merge pull request #50 from GPflow/tf_main
Tip revision: c0f0fbe
test_mean_functions.py
import GPflow
import tensorflow as tf
import numpy as np
import unittest
class TestMeanFuncs(unittest.TestCase):
def setUp(self):
self.input_dim=3
self.output_dim=2
self.N=20
rng = np.random.RandomState(0)
self.mfs = [ GPflow.mean_functions.Zero(),
GPflow.mean_functions.Linear(rng.randn(self.input_dim, self.output_dim), rng.randn(self.output_dim)),
GPflow.mean_functions.Constant(rng.randn(self.output_dim))]
self.x = tf.placeholder('float64')
for mf in self.mfs:
mf.make_tf_array(self.x)
self.X = tf.placeholder(tf.float64, [self.N, self.input_dim])
self.X_data = np.random.randn(self.N, self.input_dim)
def test_output_shape(self):
for mf in self.mfs:
with mf.tf_mode():
Y = tf.Session().run(mf(self.X), feed_dict={self.x: mf.get_free_state(), self.X:self.X_data})
self.failUnless(Y.shape in [(self.N, self.output_dim), (self.N, 1)])
class TestModelsWithMeanFuncs(unittest.TestCase):
"""
Simply check that all models have a higher prediction with a constant mean
function than with a zero mean function.
"""
def setUp(self):
self.input_dim=3
self.output_dim=2
self.N=20
self.Ntest=30
self.M=5
rng = np.random.RandomState(0)
X, Y, Z, self.Xtest = rng.randn(self.N, self.input_dim),\
rng.randn(self.N, self.output_dim),\
rng.randn(self.M, self.input_dim),\
rng.randn(self.Ntest, self.input_dim)
k = lambda : GPflow.kernels.Matern32(self.input_dim)
zero = GPflow.mean_functions.Zero()
const = GPflow.mean_functions.Constant(np.ones(self.output_dim) * 10)
self.models_with, self.models_without =\
[[GPflow.gpr.GPR(X, Y, mean_function=mf, kern=k()),
GPflow.sgpr.SGPR(X, Y, mean_function=mf, Z=Z, kern=k()),
GPflow.sgpr.GPRFITC(X, Y, mean_function=mf, Z=Z, kern=k()),
GPflow.svgp.SVGP(X, Y, mean_function=mf, Z=Z, kern=k(), likelihood=GPflow.likelihoods.Gaussian()),
GPflow.vgp.VGP(X, Y, mean_function=mf, kern=k(), likelihood=GPflow.likelihoods.Gaussian()),
GPflow.vgp.VGP(X, Y, mean_function=mf, kern=k(), likelihood=GPflow.likelihoods.Gaussian()),
GPflow.gpmc.GPMC(X, Y, mean_function=mf, kern=k(), likelihood=GPflow.likelihoods.Gaussian()),
GPflow.sgpmc.SGPMC(X, Y, mean_function=mf, kern=k(), likelihood=GPflow.likelihoods.Gaussian(), Z=Z)] for mf in (const, zero)]
def test_mean_function(self):
for m_with, m_without in zip(self.models_with, self.models_without):
mu1, v1 = m_with.predict_f(self.Xtest)
mu2, v2 = m_without.predict_f(self.Xtest)
self.failUnless(np.all(v1==v2))
self.failIf(np.all(mu1 == mu2))
if __name__ == "__main__":
unittest.main()
Computing file changes ...