diff options
Diffstat (limited to 'ml_exp/__init__.py')
-rw-r--r-- | ml_exp/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index c62141a58..48ece56c8 100644 --- a/ml_exp/__init__.py +++ b/ml_exp/__init__.py @@ -25,6 +25,7 @@ from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\ first_neighbor_matrix, adjacency_matrix, check_bond, bag_of_stuff from ml_exp.math import cholesky_solve from ml_exp.qm7db import qm7db +from ml_exp.do_ml import simple_ml, do_ml __all__ = ['Compound', 'coulomb_matrix', @@ -34,4 +35,6 @@ __all__ = ['Compound', 'check_bond', 'bag_of_stuff', 'cholesky_solve', - 'qm7db'] + 'qm7db', + 'simple_ml', + 'do_ml'] |