diff --git a/GPy/kern/src/coregionalize.py b/GPy/kern/src/coregionalize.py index d05f5c6a..a5a72bb0 100644 --- a/GPy/kern/src/coregionalize.py +++ b/GPy/kern/src/coregionalize.py @@ -150,6 +150,7 @@ class Coregionalize(Kern): input_dict["W"] = self.W.values.tolist() input_dict["kappa"] = self.kappa.values.tolist() input_dict["output_dim"] = self.output_dim + input_dict["rank"] = self.rank return input_dict @staticmethod diff --git a/GPy/util/multioutput.py b/GPy/util/multioutput.py index 91227838..0b1772e2 100644 --- a/GPy/util/multioutput.py +++ b/GPy/util/multioutput.py @@ -75,7 +75,7 @@ def ICM(input_dim, num_outputs, kernel, W_rank=1,W=None,kappa=None,name='ICM'): :num_outputs: Number of outputs :param kernel: kernel that will be multiplied by the coregionalize kernel (matrix B). :type kernel: a GPy kernel - :param W_rank: number tuples of the corregionalization parameters 'W' + :param W_rank: number tuples of the coregionalization parameters 'W' :type W_rank: integer """ if kernel.input_dim != input_dim: