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

[newest patch updates, cleaned interfaces and mean_function addidtions]

Browse source
This commit is contained in:
mzwiessele 2015-09-30 14:43:04 +01:00
parent 75ccd468ef
commit 71bc90bf98
6 changed files with 28 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

4
GPy/models/gp_classification.py
View file

@ -20,10 +20,10 @@ class GPClassification(GP):
"""
def __init__(self, X, Y, kernel=None,Y_metadata=None):
def __init__(self, X, Y, kernel=None,Y_metadata=None, mean_function=None):
if kernel is None:
kernel = kern.RBF(X.shape[1])
likelihood = likelihoods.Bernoulli()
GP.__init__(self, X=X, Y=Y, kernel=kernel, likelihood=likelihood, inference_method=EP(), name='gp_classification')
GP.__init__(self, X=X, Y=Y, kernel=kernel, likelihood=likelihood, inference_method=EP(), mean_function=mean_function, name='gp_classification')