mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-06-05 14:55:15 +02:00
removed climin dependency unless actually needed
This commit is contained in:
parent
06d7e690f3
commit
965f1aa2cb
1 changed files with 13 additions and 12 deletions
|
|
@ -25,14 +25,15 @@ class SVGP(SparseGP):
|
|||
|
||||
Hensman, Matthews and Ghahramani, Scalable Variational GP Classification, ArXiv 1411.2005
|
||||
"""
|
||||
self.batchsize = batchsize
|
||||
if batchsize is None:
|
||||
batchsize = X.shape[0]
|
||||
X_batch, Y_batch = X, Y
|
||||
KL_scale, batch_scale = 1., 1.
|
||||
|
||||
else:
|
||||
self.X_all, self.Y_all = X, Y
|
||||
# how to rescale the batch likelihood in case of minibatches
|
||||
self.batchsize = batchsize
|
||||
batch_scale = float(self.X_all.shape[0])/float(self.batchsize)
|
||||
#KL_scale = 1./np.float64(self.mpi_comm.size)
|
||||
KL_scale = 1.0
|
||||
|
||||
import climin.util
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue