apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-06-08 15:05:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

saving W_rank variable in from_dict method for Coregionalization kernels + typo fix

Browse source
This commit is contained in:
gehbiszumeis 2022-02-23 16:19:49 +01:00
parent aa9a2ee360
commit f161325f62
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
GPy/kern/src/coregionalize.py
View file

@ -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

2
GPy/util/multioutput.py
View file

@ -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: