summaryrefslogtreecommitdiff
path: root/ml_exp/__main__.py
diff options
context:
space:
mode:
authorDavid Luevano Alvarado <55825613+luevano@users.noreply.github.com>2020-02-23 21:03:39 -0700
committerDavid Luevano Alvarado <55825613+luevano@users.noreply.github.com>2020-02-23 21:03:39 -0700
commitdb47ff8606b14e4cdddc0e317972c3515b9c788b (patch)
tree8f3d23c64a82a5f580c51fdd55e2ae070957e83b /ml_exp/__main__.py
parent85198178482767d2831ddf3762e5e6eeafa990bd (diff)
Fix adj matrix issue with cc bonds)
Diffstat (limited to 'ml_exp/__main__.py')
-rw-r--r--ml_exp/__main__.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/ml_exp/__main__.py b/ml_exp/__main__.py
index 17119623e..124ce8cd9 100644
--- a/ml_exp/__main__.py
+++ b/ml_exp/__main__.py
@@ -25,6 +25,11 @@ from ml_exp.compound import Compound
if __name__ == '__main__':
print('Initialize test')
- test = []
- for i in range(1, 10):
- test.append(Compound(f'/home/luevano/py/ml_exp/data/qm7/000{i}.xyz'))
+ mol_name = '/home/luevano/py/ml_exp/data/qm7/0004.xyz'
+ mol = Compound(mol_name)
+ mol.gen_cm(size=1, as_eig=False)
+ mol.gen_ljm(size=1, as_eig=False)
+ mol.gen_am(size=1)
+
+ print(mol.n, mol.atoms, mol.atoms_nc)
+ print(mol.am)