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

fix pickle

Browse source
This commit is contained in:
Zhenwen Dai 2014-05-16 11:23:44 +01:00
parent 17b6b94db3
commit e6d07ad5ac
7 changed files with 230 additions and 275 deletions
Download patch file Download diff file Expand all files Collapse all files

7
GPy/inference/latent_function_inference/var_dtc_parallel.py
View file

@ -26,7 +26,7 @@ class VarDTC_minibatch(LatentFunctionInference):
"""
const_jitter = 1e-6
def __init__(self, batchsize, limit=1, mpi_comm=None):
def __init__(self, batchsize=None, limit=1, mpi_comm=None):
self.batchsize = batchsize
self.mpi_comm = mpi_comm
@ -74,9 +74,12 @@ class VarDTC_minibatch(LatentFunctionInference):
return jitchol(tdot(Y))
def gatherPsiStat(self, kern, X, Z, Y, beta, uncertain_inputs, het_noise):
num_inducing = Z.shape[0]
num_data, output_dim = Y.shape
if self.batchsize == None or self.batchsize>num_data:
self.batchsize = num_data
trYYT = self.get_trYYT(Y)