mean functions in place

This commit is contained in:
James Hensman 2015-03-23 14:47:49 +00:00
parent efb32282de
commit fa801bf46c
8 changed files with 18 additions and 9 deletions

View file

@ -6,7 +6,8 @@ from posterior import Posterior
class SVGP(LatentFunctionInference):
def inference(self, q_u_mean, q_u_chol, kern, X, Z, likelihood, Y, Y_metadata=None, KL_scale=1.0, batch_scale=1.0):
def inference(self, q_u_mean, q_u_chol, kern, X, Z, likelihood, Y, mean_function=None, Y_metadata=None, KL_scale=1.0, batch_scale=1.0):
assert mean_function is None, "inference with a mean function not implemented"
num_inducing = Z.shape[0]
num_data, num_outputs = Y.shape