diff options
Diffstat (limited to 'ml_exp/__main__.py')
| -rw-r--r-- | ml_exp/__main__.py | 11 |
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) |