Fixed an assertion, it was checking the dimensionality of the input data matrix, rather than that of the labels.

This commit is contained in:
teodecampos 2014-05-23 17:18:38 +01:00
parent 0252fae298
commit f84f3f4308

View file

@ -32,7 +32,7 @@ class EP(LatentFunctionInference):
pass
def inference(self, kern, X, likelihood, Y, Y_metadata=None, Z=None):
num_data, output_dim = X.shape
num_data, output_dim = Y.shape
assert output_dim ==1, "ep in 1D only (for now!)"
K = kern.K(X)