Update KISSGP_Regression_1D_CUDA.ipynb
" def __init__(self, train_x, train_y, likelihood):\n",
" super(GPRegressionModel, self).__init__(train_x, train_y, likelihood)\n",
" \n",
- " # SKI requires a grid size hyperparameter. This util can help with that\n",
- " grid_size = gpytorch.utils.grid.choose_grid_size(train_x)\n",
+ " # SKI requires a grid size hyperparameter. This util can help with that. Here we are using the same number of grid points as training points (a ratio of 1). Performance can be sensitive to this parameter (the number of grid points), so you may want to adjust it to your problem on a validation set.\n",
+ " grid_size = gpytorch.utils.grid.choose_grid_size(train_x,1.0)\n",
" \n",
" self.mean_module = gpytorch.means.ConstantMean()\n",
" self.covar_module = gpytorch.kernels.GridInterpolationKernel(\n",
