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
|
Hensman, Matthews and Ghahramani, Scalable Variational GP Classification, ArXiv 1411.2005
|
||||||
"""
|
"""
|
||||||
|
self.batchsize = batchsize
|
||||||
if batchsize is None:
|
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
|
self.X_all, self.Y_all = X, Y
|
||||||
# how to rescale the batch likelihood in case of minibatches
|
# how to rescale the batch likelihood in case of minibatches
|
||||||
self.batchsize = batchsize
|
|
||||||
batch_scale = float(self.X_all.shape[0])/float(self.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
|
KL_scale = 1.0
|
||||||
|
|
||||||
import climin.util
|
import climin.util
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue