diff --git a/deepgp/layers/mrd.py b/deepgp/layers/mrd.py index 805b2b0..f4b89d4 100644 --- a/deepgp/layers/mrd.py +++ b/deepgp/layers/mrd.py @@ -108,8 +108,9 @@ def __init__(self, dim_down, dim_up, likelihood, MLP_dims=None, X=None, X_varian #assert back_constraint self.uncertain_inputs = uncertain_inputs - Y = self.Y if self.layer_lower is None else self.layer_lower.X + Y = self.Y if self.layer_lower is None else self.layer_lower.X assert isinstance(dim_down, list) or isinstance(dim_down, tuple) + assert isinstance(kernel, list) and len(kernel)==len(dim_down), "The number of kernels has to be equal to the number of input modalities!" super(MRDLayer, self).__init__(name=name) self.mpi_comm, self.mpi_root = mpi_comm, mpi_root diff --git a/deepgp/models/model.py b/deepgp/models/model.py index c4ba672..b9101f4 100644 --- a/deepgp/models/model.py +++ b/deepgp/models/model.py @@ -8,7 +8,7 @@ from GPy import Model,likelihoods from GPy.core.parameterization.variational import VariationalPosterior,\ NormalPosterior -from ..layers import ObservedLayer, HiddenLayer, TopHiddenLayer +from ..layers import ObservedLayer, ObservedMRDLayer, HiddenLayer, TopHiddenLayer class DeepGP(Model):