mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-08 11:32:39 +02:00
Working One vs All sparse gp classification wrapper
This commit is contained in:
parent
031aa7b315
commit
aa473178f7
1 changed files with 5 additions and 3 deletions
|
|
@ -18,9 +18,9 @@ class OneVsAllSparseClassification(object):
|
|||
|
||||
"""
|
||||
|
||||
def __init__(self, X, Y, kernel=None,Y_metadata=None,messages=True):
|
||||
def __init__(self, X, Y, kernel=None,Y_metadata=None,messages=True,num_inducing=10):
|
||||
if kernel is None:
|
||||
kernel = GPy.kern.RBF(X.shape[1])
|
||||
kernel = GPy.kern.RBF(X.shape[1]) + GPy.kern.White(X.shape[1]) + GPy.kern.Bias(X.shape[1])
|
||||
|
||||
likelihood = GPy.likelihoods.Bernoulli()
|
||||
|
||||
|
|
@ -35,6 +35,8 @@ class OneVsAllSparseClassification(object):
|
|||
Ynew[Y.flatten()!=yj] = 0
|
||||
Ynew[Y.flatten()==yj] = 1
|
||||
|
||||
m = GPy.models.SparseGPClassification(X,Ynew,kernel=kernel,Y_metadata=Y_metadata)
|
||||
m = GPy.models.SparseGPClassification(X,Ynew,kernel=kernel.copy(),Y_metadata=Y_metadata,num_inducing=num_inducing)
|
||||
m.optimize(messages=messages)
|
||||
self.results[yj] = m.predict(X)[0]
|
||||
del m
|
||||
del Ynew
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue