b61a525baee25d2ea31ab088540dd4d2ffed2251,test/examples/simple_gp_regression_test.py,ExactGPModel,__init__,#ExactGPModel#,21

Before Change

        self.mean_module = ConstantMean()
        self.covar_module = RBFKernel()

        self.register_parameter("constant_mean", nn.Parameter(torch.Tensor([0])), bounds=(-1, 1))
        self.register_parameter("log_noise", nn.Parameter(torch.Tensor([0])), bounds=(-5, 5))
        self.register_parameter("log_lengthscale", nn.Parameter(torch.Tensor([0])), bounds=(-5, 5))

    def forward(self, x):

After Change

class ExactGPModel(gpytorch.GPModel):
    def __init__(self):
        likelihood = GaussianLikelihood(log_noise_bounds=(-3, 3))
        super(ExactGPModel, self).__init__(likelihood)
        self.mean_module = ConstantMean(constant_bounds=(-1, 1))
        self.covar_module = RBFKernel(log_lengthscale_bounds=(-3, 3))

Instances