Merge pull request #1410 from cornellius-gp/multitask-dgp-example

Add multitask DGP example.
cornellius-gp · Jan 19, 2021 · f1ae39d · f1ae39d
2 parents 2b3014a + 6982cbf
commit f1ae39d
Show file tree

Hide file tree

Showing 5 changed files with 312 additions and 8 deletions.
diff --git a/examples/04_Variational_and_Approximate_GPs/SVGP_Multitask_GP_Regression.ipynb b/examples/04_Variational_and_Approximate_GPs/SVGP_Multitask_GP_Regression.ipynb
@@ -360,7 +360,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.3"
+   "version": "3.7.0"
   }
  },
  "nbformat": 4,

diff --git a/examples/05_Deep_Gaussian_Processes/DGP_Multitask_Regression.ipynb b/examples/05_Deep_Gaussian_Processes/DGP_Multitask_Regression.ipynb
diff --git a/examples/05_Deep_Gaussian_Processes/Deep_Gaussian_Processes.ipynb b/examples/05_Deep_Gaussian_Processes/Deep_Gaussian_Processes.ipynb
@@ -31,7 +31,6 @@
     "import torch\n",
     "import tqdm\n",
     "import gpytorch\n",
-    "from torch.nn import Linear\n",
     "from gpytorch.means import ConstantMean, LinearMean\n",
     "from gpytorch.kernels import RBFKernel, ScaleKernel\n",
     "from gpytorch.variational import VariationalStrategy, CholeskyVariationalDistribution\n",
@@ -169,10 +168,8 @@
     "            batch_shape=batch_shape, ard_num_dims=None\n",
     "        )\n",
     "\n",
-    "        self.linear_layer = Linear(input_dims, 1)\n",
-    "\n",
     "    def forward(self, x):\n",
-    "        mean_x = self.mean_module(x) # self.linear_layer(x).squeeze(-1)\n",
+    "        mean_x = self.mean_module(x)\n",
     "        covar_x = self.covar_module(x)\n",
     "        return MultivariateNormal(mean_x, covar_x)\n",
     "\n",
@@ -544,7 +541,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.3"
+   "version": "3.7.0"
   }
  },
  "nbformat": 4,

diff --git a/examples/05_Deep_Gaussian_Processes/Deep_Sigma_Point_Processes.ipynb b/examples/05_Deep_Gaussian_Processes/Deep_Sigma_Point_Processes.ipynb
@@ -24,7 +24,6 @@
     "from gpytorch.kernels import ScaleKernel, MaternKernel\n",
     "from gpytorch.variational import VariationalStrategy, BatchDecoupledVariationalStrategy\n",
     "from gpytorch.variational import MeanFieldVariationalDistribution\n",
-    "from gpytorch.models.deep_gps import DeepGP\n",
     "from gpytorch.models.deep_gps.dspp import DSPPLayer, DSPP\n",
     "import gpytorch.settings as settings"
    ]
@@ -487,7 +486,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.7.0"
   }
  },
  "nbformat": 4,

diff --git a/examples/05_Deep_Gaussian_Processes/index.rst b/examples/05_Deep_Gaussian_Processes/index.rst
@@ -5,3 +5,4 @@
 
    Deep_Gaussian_Processes.ipynb
    Deep_Sigma_Point_Processes.ipynb
+   DGP_Multitask_Regression.ipynb
Original file line number	Diff line number	Diff line change
Expand Up		@@ -5,3 +5,4 @@

		Deep_Gaussian_Processes.ipynb
		Deep_Sigma_Point_Processes.ipynb
		DGP_Multitask_Regression.ipynb