diff --git a/GPy/util/initialization.py b/GPy/util/initialization.py index 22e63b6b..8d23b541 100644 --- a/GPy/util/initialization.py +++ b/GPy/util/initialization.py @@ -13,7 +13,11 @@ def initialize_latent(init, input_dim, Y): p = pca(Y) PC = p.project(Y, min(input_dim, Y.shape[1])) Xr[:PC.shape[0], :PC.shape[1]] = PC + vars = p.fracs[:input_dim] else: - var = Xr.var(0) - return Xr, var/var.max() - return Xr, p.fracs[:input_dim] \ No newline at end of file + vars = Xr.var(0) + + Xr -= Xr.mean(0) + Xr /= Xr.var(0) + + return Xr, vars/vars.max() \ No newline at end of file