fixed imports

cambridge-mlg · Oct 16, 2018 · 8c6398c · 8c6398c
1 parent fe95847
commit 8c6398c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/conditionals.py b/conditionals.py
@@ -1,7 +1,6 @@
 # Credit to GPflow
 
 import tensorflow as tf
-import numpy as np
 
 
 def base_conditional(Kmn, Kmm, Knn, f, *, full_cov=False, q_sqrt=None, white=False):
@@ -98,7 +97,7 @@ def conditional(Xnew, X, kern, f, *, full_cov=False, q_sqrt=None, white=False):
     """
 
     num_data = tf.shape(X)[0]  # M
-    Kmm = kern.K(X) + tf.eye(num_data, dtype=tf.float64) * 1e-10
+    Kmm = kern.K(X) + tf.eye(num_data, dtype=tf.float64) * 1e-7
     Kmn = kern.K(X, Xnew)
     if full_cov:
         Knn = kern.K(Xnew)

diff --git a/models.py b/models.py
@@ -4,8 +4,6 @@
 import numpy as np
 from kernels import SquaredExponential
 from likelihoods import Gaussian
-from scipy.cluster.vq import kmeans2
-import tensorflow as tf
 
 class RegressionModel(object):
     def __init__(self, is_test=False, seed=0):