apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-04-30 23:36:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Changes to allow mixed noise likelihoods

Browse source
This commit is contained in:
Ricardo 2014-03-17 19:00:16 +00:00
parent bea74ac523
commit 385ce7d344
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
GPy/plotting/matplot_dep/models_plots.py
View file

@ -7,6 +7,7 @@ import Tango
from base_plots import gpplot, x_frame1D, x_frame2D
from ...util.misc import param_to_array
from ...models.gp_coregionalized_regression import GPCoregionalizedRegression
from ...models.sparse_gp_coregionalized_regression import SparseGPCoregionalizedRegression
def plot_fit(model, plot_limits=None, which_data_rows='all',
@ -86,7 +87,10 @@ def plot_fit(model, plot_limits=None, which_data_rows='all',
lower = m - 2*np.sqrt(v)
upper = m + 2*np.sqrt(v)
else:
meta = {'output_index': Xgrid[:,-1:].astype(np.int)} if isinstance(model,GPCoregionalizedRegression) else None
if isinstance(model,GPCoregionalizedRegression) or isinstance(model,SparseGPCoregionalizedRegression):
meta = {'output_index': Xgrid[:,-1:].astype(np.int)}
else:
meta = None
m, v = model.predict(Xgrid, full_cov=False, Y_metadata=meta)
lower, upper = model.predict_quantiles(Xgrid, Y_metadata=meta)