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

[vardtc missing data] can handle non broadcastable selections

Browse source
This commit is contained in:
Max Zwiessele 2014-05-20 14:45:34 +01:00
parent 8c8d06c8ae
commit 99699e9e02
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
GPy/inference/latent_function_inference/var_dtc.py
View file

@ -283,7 +283,11 @@ class VarDTCMissingData(LatentFunctionInference):
else: beta = beta_all else: beta = beta_all
VVT_factor = (beta*y) VVT_factor = (beta*y)
VVT_factor_all[v, ind].flat = VVT_factor.flat try:
VVT_factor_all[v, ind].flat = VVT_factor.flat
except ValueError:
mult = np.ravel_multi_index((v.nonzero()[0][:,None],ind[None,:]), VVT_factor_all.shape)
VVT_factor_all.flat[mult] = VVT_factor
output_dim = y.shape[1] output_dim = y.shape[1]
psi0 = psi0_all[v] psi0 = psi0_all[v]

2
GPy/util/subarray_and_sorting.py
View file

@ -4,7 +4,7 @@
.. moduleauthor:: Max Zwiessele <ibinbei@gmail.com> .. moduleauthor:: Max Zwiessele <ibinbei@gmail.com>
''' '''
__updated__ = '2013-12-02' __updated__ = '2014-05-20'
import numpy as np import numpy as np