bce6052bd84a421a44bc590b1eda23923f63b951,test/examples/kissgp_additive_classification_test.py,GPClassificationModel,__init__,#GPClassificationModel#,35

Before Change

class GPClassificationModel(gpytorch.GPModel):
    def __init__(self):
        super(GPClassificationModel, self).__init__(BernoulliLikelihood())
        self.latent_function = LatentFunction()

    def forward(self, x):
        return self.latent_function(x)

After Change

class GPClassificationModel(gpytorch.models.AdditiveGridInducingVariationalGP):
    def __init__(self):
        super(GPClassificationModel, self).__init__(grid_size=16, grid_bounds=[(-1, 1)],
                                                    n_components=2, mixing_params=True)
        self.mean_module = ConstantMean(constant_bounds=[-1e-5, 1e-5])
        self.covar_module = RBFKernel(log_lengthscale_bounds=(-5, 6))
        self.register_parameter("log_outputscale", nn.Parameter(torch.Tensor([0])), bounds=(-5, 6))

    def forward(self, x):
        mean_x = self.mean_module(x)
        covar_x = self.covar_module(x)

Instances