From 4c3c2ac755cb5c0b4e93d93f1806ee2bc9f814f4 Mon Sep 17 00:00:00 2001 From: gehbiszumeis <16896724+gehbiszumeis@users.noreply.github.com> Date: Fri, 29 Oct 2021 13:32:16 +0200 Subject: [PATCH] added additional needed parameters to to_dict() method for Coregionalize kernel + added _build_from_input dict method --- GPy/kern/src/coregionalize.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/GPy/kern/src/coregionalize.py b/GPy/kern/src/coregionalize.py index 41ddf841..d05f5c6a 100644 --- a/GPy/kern/src/coregionalize.py +++ b/GPy/kern/src/coregionalize.py @@ -146,6 +146,16 @@ class Coregionalize(Kern): input_dict = super(Coregionalize, self)._save_to_input_dict() input_dict["class"] = "GPy.kern.Coregionalize" + # W and kappa must be serializable input_dict["W"] = self.W.values.tolist() input_dict["kappa"] = self.kappa.values.tolist() + input_dict["output_dim"] = self.output_dim return input_dict + + @staticmethod + def _build_from_input_dict(kernel_class, input_dict): + useGPU = input_dict.pop('useGPU', None) + # W and kappa must be converted back to numpy arrays + input_dict['W'] = np.array(input_dict['W']) + input_dict['kappa'] = np.array(input_dict['kappa']) + return Coregionalize(**input_dict)