15b3563c3c7f700ee0bf23b0477527dfa5323cea,test/examples/test_kissgp_additive_regression.py,GPRegressionModel,__init__,#GPRegressionModel#Any#Any#Any#,41

Before Change

    def __init__(self, train_x, train_y, likelihood):
        super(GPRegressionModel, self).__init__(train_x, train_y, likelihood)
        self.mean_module = ConstantMean(constant_bounds=(-1, 1))
        self.base_covar_module = RBFKernel(log_lengthscale_bounds=(-3, 3))
        self.covar_module = AdditiveGridInterpolationKernel(
            self.base_covar_module, grid_size=100, grid_bounds=[(0, 1)], n_components=2
        )

After Change

    def __init__(self, train_x, train_y, likelihood):
        super(GPRegressionModel, self).__init__(train_x, train_y, likelihood)
        self.mean_module = ConstantMean(prior=SmoothedBoxPrior(-1, 1))
        self.base_covar_module = RBFKernel(
            log_lengthscale_prior=SmoothedBoxPrior(exp(-3), exp(3), sigma=0.1, log_transform=True)
        )
        self.covar_module = AdditiveGridInterpolationKernel(
            self.base_covar_module, grid_size=100, grid_bounds=[(0, 1)], n_components=2
        )

Instances