summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDavid Luevano Alvarado <55825613+luevano@users.noreply.github.com>2020-03-26 15:53:27 -0700
committerDavid Luevano Alvarado <55825613+luevano@users.noreply.github.com>2020-03-26 15:53:27 -0700
commit1d7ea44f8b5dba01ad2660b8014002b989dff464 (patch)
tree9163d95388e32a6bee37af20c18e3b0b9db22779
parentf8e5350630da8b424a8a5f44a8166c26648d3dac (diff)
Fix kernels
-rw-r--r--ml_exp/krr.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/ml_exp/krr.py b/ml_exp/krr.py
index 60335fa77..4a87884d4 100644
--- a/ml_exp/krr.py
+++ b/ml_exp/krr.py
@@ -30,7 +30,8 @@ except ImportError:
print('Tensorflow couldn\'t be imported. Maybe it is not installed.')
TF_AV = False
from ml_exp.misc import printc
-from ml_exp.kernels import laplauss_kernel
+from ml_exp.kernels import gaussian_kernel, laplacian_kernel,\
+ wasserstein_kernel
from ml_exp.readdb import qm7db