diff options
Diffstat (limited to 'ml_exp/__init__.py')
-rw-r--r-- | ml_exp/__init__.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index 6e901ec2c..d304f581c 100644 --- a/ml_exp/__init__.py +++ b/ml_exp/__init__.py @@ -22,8 +22,9 @@ SOFTWARE. """ from ml_exp.compound import Compound from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\ - get_helping_data, adjacency_matrix, check_bond, bag_of_bonds + get_helping_data, adjacency_matrix, epsilon_index, check_bond, bag_of_bonds 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 @@ -32,9 +33,12 @@ __all__ = ['Compound', 'lennard_jones_matrix', 'get_helping_data', 'adjacency_matrix', + 'epsilon_index', 'check_bond', 'bag_of_bonds', 'qm7db', 'laplauss_kernel', 'simple_ml', - 'do_ml'] + 'do_ml', + 'NUCLEAR_CHARGE', + 'POSSIBLE_BONDS'] |