apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-08 11:32:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'devel' of github.com:SheffieldML/GPy into devel

Browse source
This commit is contained in:
Ricardo 2014-11-14 21:24:31 +00:00
commit 031aa7b315
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
GPy/models/one_vs_all_classification.py
View file

@ -2,11 +2,13 @@
# Licensed under the BSD 3-clause license (see LICENSE.txt)
from ..core import GP
from . import SparseGPClassification
from .. import likelihoods
from .. import kern
from ..inference.latent_function_inference.expectation_propagation import EP
import numpy as np
class OneVsAllClassification(GP):
class OneVsAllClassification(object):
"""
Gaussian Process classification: One vs all
@ -20,7 +22,7 @@ class OneVsAllClassification(GP):
"""
def __init__(self, X, Y, kernel=None,Y_metadata=None):
def __init__(self, X, Y, kernel=None,Y_metadata=None,messages=True):
if kernel is None:
kernel = kern.RBF(X.shape[1])
@ -36,6 +38,7 @@ class OneVsAllClassification(GP):
Ynew[Y.flatten()!=yj] = 0
Ynew[Y.flatten()==yj] = 1
m = GPy.models.GPClassification(X,Ynew,kernel=kernel,Y_metadata=Y_metadata,inference_method=inference_method)
m.optimize()
m = SparseGPClassification(X,Ynew,kernel=kernel,Y_metadata=Y_metadata)
m.optimize(messages=messages)
stop
self.results[yj] = m.predict(X)