diff options
Diffstat (limited to 'ml_exp/__init__.py')
-rw-r--r-- | ml_exp/__init__.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index d304f581c..02819d6a1 100644 --- a/ml_exp/__init__.py +++ b/ml_exp/__init__.py @@ -26,7 +26,7 @@ from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\ from ml_exp.qm7db import qm7db from ml_exp.data import NUCLEAR_CHARGE, POSSIBLE_BONDS from ml_exp.kernels import laplauss_kernel -from ml_exp.do_ml import simple_ml, do_ml +from ml_exp.krr import krr, multi_krr __all__ = ['Compound', 'coulomb_matrix', @@ -38,7 +38,7 @@ __all__ = ['Compound', 'bag_of_bonds', 'qm7db', 'laplauss_kernel', - 'simple_ml', - 'do_ml', + 'krr', + 'multi_krr', 'NUCLEAR_CHARGE', 'POSSIBLE_BONDS'] |