diff --git a/GPy/examples/classification.py b/GPy/examples/classification.py index f3adebaa..421440b3 100644 --- a/GPy/examples/classification.py +++ b/GPy/examples/classification.py @@ -62,7 +62,7 @@ def oil(): likelihood = GPy.likelihoods.EP(Y, distribution) # Create GP model - m = GPy.models.GP(data['X'], likelihood=likelihood, kernel=kernel) + m = GPy.models.GP_classification(data['X'], Y, kernel=kernel) # Contrain all parameters to be positive m.constrain_positive('') @@ -93,9 +93,11 @@ def toy_linear_1d_classification(seed=default_seed): link = GPy.likelihoods.link_functions.probit distribution = GPy.likelihoods.likelihood_functions.binomial(link) likelihood = GPy.likelihoods.EP(Y, distribution) + Y[1] = 1 # Model definition - m = GPy.models.GP(data['X'], likelihood=likelihood, kernel=kernel) + #m = GPy.models.GP(data['X'], likelihood=likelihood, kernel=kernel) + m = GPy.models.GP_classification(data['X'], Y, likelihood=likelihood, kernel=kernel) m.ensure_default_constraints() # Optimize diff --git a/GPy/models/__init__.py b/GPy/models/__init__.py index 700ea120..676153c7 100644 --- a/GPy/models/__init__.py +++ b/GPy/models/__init__.py @@ -5,6 +5,7 @@ #from GP import GP #from sparse_GP import sparse_GP from GP_regression import GP_regression +from GP_classification import GP_classification from sparse_GP_regression import sparse_GP_regression from GPLVM import GPLVM from warped_GP import warpedGP