apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-27 14:25:16 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

(wpgs) fixing newton-raphson for f_inv and fixing plotting stuff

Browse source
This commit is contained in:
beckdaniel 2015-12-08 13:59:46 +00:00
parent caa962069d
commit bef114eabd
4 changed files with 50 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

8
GPy/plotting/gpy_plot/plot_util.py
View file

@ -31,6 +31,7 @@
import numpy as np
from scipy import sparse
import itertools
from GPy.models import WarpedGP
def in_ipynb():
try:
@ -73,6 +74,9 @@ def helper_predict_with_model(self, Xgrid, plot_raw, apply_link, percentiles, wh
if 'output_index' not in predict_kw['Y_metadata']:
predict_kw['Y_metadata']['output_index'] = Xgrid[:,-1:].astype(np.int)
if isinstance(self, WarpedGP) and self.predict_in_warped_space:
predict_kw['median'] = True
mu, _ = self.predict(Xgrid, **predict_kw)
if percentiles is not None:
@ -291,6 +295,8 @@ def get_x_y_var(model):
Y = model.Y.values
except AttributeError:
Y = model.Y
if isinstance(model, WarpedGP) and model.predict_in_warped_space:
Y = model.Y_untransformed
if sparse.issparse(Y): Y = Y.todense().view(np.ndarray)
return X, X_variance, Y
@ -377,4 +383,4 @@ def x_frame2D(X,plot_limits=None,resolution=None):
resolution = resolution or 50
xx, yy = np.mgrid[xmin[0]:xmax[0]:1j*resolution,xmin[1]:xmax[1]:1j*resolution]
Xnew = np.vstack((xx.flatten(),yy.flatten())).T
return Xnew, xx, yy, xmin, xmax
return Xnew, xx, yy, xmin, xmax