diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-11 10:52:02 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-11 10:52:02 -0700 |
| commit | e94c61e224f8cd4b66a870a9d9d03c0ed810d656 (patch) | |
| tree | 27619a21eaf6c4943a2ca65f1f7a4272f0cfa9b1 /ml_exp/__init__.py | |
| parent | 25a3933873f6b7f074e190ad0256f4ecdc40b9a2 (diff) | |
Refactor db reading
Diffstat (limited to 'ml_exp/__init__.py')
| -rw-r--r-- | ml_exp/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index 02819d6a1..b40957baf 100644 --- a/ml_exp/__init__.py +++ b/ml_exp/__init__.py @@ -23,7 +23,7 @@ SOFTWARE. from ml_exp.compound import Compound 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.qm7db import qm7db +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.krr import krr, multi_krr @@ -37,6 +37,7 @@ __all__ = ['Compound', 'check_bond', 'bag_of_bonds', 'qm7db', + 'qm9db', 'laplauss_kernel', 'krr', 'multi_krr', |