diff --git a/GPy/models/gplvm.py b/GPy/models/gplvm.py index cdc0ab47..3d0b0d5c 100644 --- a/GPy/models/gplvm.py +++ b/GPy/models/gplvm.py @@ -14,7 +14,7 @@ class GPLVM(GP): """ - def __init__(self, Y, input_dim, init='PCA', X=None, kernel=None, name="gplvm"): + def __init__(self, Y, input_dim, init='PCA', X=None, kernel=None, name="gplvm", Y_metadata=None, normalizer=False): """ :param Y: observed data @@ -23,6 +23,11 @@ class GPLVM(GP): :type input_dim: int :param init: initialisation method for the latent space :type init: 'PCA'|'random' + :param normalizer: + normalize the outputs Y. + If normalizer is True, we will normalize using Standardize. + If normalizer is False (the default), no normalization will be done. + :type normalizer: bool """ if X is None: from ..util.initialization import initialize_latent @@ -34,7 +39,7 @@ class GPLVM(GP): likelihood = Gaussian() - super(GPLVM, self).__init__(X, Y, kernel, likelihood, name='GPLVM') + super(GPLVM, self).__init__(X, Y, kernel, likelihood, name='GPLVM', Y_metadata=Y_metadata, normalizer=normalizer) self.X = Param('latent_mean', X) self.link_parameter(self.X, index=0)