summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ml_exp/__init__.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py
index b40957baf..e1d3bcd6f 100644
--- a/ml_exp/__init__.py
+++ b/ml_exp/__init__.py
@@ -25,7 +25,7 @@ from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\
get_helping_data, adjacency_matrix, epsilon_index, check_bond, bag_of_bonds
from ml_exp.readdb import qm7db, qm9db
from ml_exp.data import NUCLEAR_CHARGE, POSSIBLE_BONDS
-from ml_exp.kernels import laplauss_kernel
+from ml_exp.kernels import gaussian_kernel, laplauss_kernel, wasserstein_kernel
from ml_exp.krr import krr, multi_krr
__all__ = ['Compound',
@@ -38,7 +38,9 @@ __all__ = ['Compound',
'bag_of_bonds',
'qm7db',
'qm9db',
- 'laplauss_kernel',
+ 'gaussian_kernel',
+ 'laplacian_kernel',
+ 'wasserstein_kernel',
'krr',
'multi_krr',
'NUCLEAR_CHARGE',