mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-04-27 13:56:23 +02:00
19 lines
526 B
Python
19 lines
526 B
Python
# Copyright (c) 2013, Alan Saul
|
|
|
|
from GPy.models import GP
|
|
from .. import likelihoods
|
|
from GPy import kern
|
|
|
|
|
|
class cox_GP_regression(GP):
|
|
"""
|
|
Cox Gaussian Process model for regression
|
|
"""
|
|
|
|
def __init__(self,X,Y,kernel=None,normalize_X=False,normalize_Y=False, Xslices=None):
|
|
if kernel is None:
|
|
kernel = kern.rbf(X.shape[1])
|
|
|
|
likelihood = likelihoods.cox_piecewise(Y, normalize=normalize_Y)
|
|
|
|
GP.__init__(self, X, likelihood, kernel, normalize_X=normalize_X, Xslices=Xslices)
|