diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-26 15:51:27 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-26 15:51:27 -0700 |
| commit | 63e9c38beb5083e620bdbe9ef1cf5fc75cba8bc4 (patch) | |
| tree | 270814e47aeb4db184fde7b7715706e81720b3f5 | |
| parent | a2bfee6e09a85dd6b3e04bcf7672f07dce527730 (diff) | |
Fix kernels
| -rw-r--r-- | ml_exp/__init__.py | 6 |
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', |