apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-07 19:12:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Y_metadata definition changed

Browse source
This commit is contained in:
Ricardo 2014-03-17 10:27:53 +00:00
parent 7ac0689156
commit bb7c26b416
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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