[white hetero] additional check for prediction

This commit is contained in:
mzwiessele 2016-03-07 15:26:44 +00:00
parent bcb7679b5c
commit 90f29405ba

View file

@ -104,7 +104,7 @@ class WhiteHeteroscedastic(Static):
return 0.
def K(self, X, X2=None):
if X2 is None:
if X2 is None and X.shape[0]==self.variance.shape[0]:
return np.eye(X.shape[0])*self.variance
else:
return np.zeros((X.shape[0], X2.shape[0]))