[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. return 0.
def K(self, X, X2=None): 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 return np.eye(X.shape[0])*self.variance
else: else:
return np.zeros((X.shape[0], X2.shape[0])) return np.zeros((X.shape[0], X2.shape[0]))