diff options
| author | David Luevano <55825613+luevano@users.noreply.github.com> | 2020-01-24 09:16:19 -0700 |
|---|---|---|
| committer | David Luevano <55825613+luevano@users.noreply.github.com> | 2020-01-24 09:16:19 -0700 |
| commit | 7494d56888d3d8eda8f5b2c732c3685ff8006ff9 (patch) | |
| tree | fc4524b26e169a5e4eedc530e76858383d0ff6e9 /ml_exp/__main__.py | |
| parent | 8dd406df884d4eb1b07a7aeb7c2a0d6d1adfe5ee (diff) | |
Add special values and force values
Diffstat (limited to 'ml_exp/__main__.py')
| -rw-r--r-- | ml_exp/__main__.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ml_exp/__main__.py b/ml_exp/__main__.py index 63708e75c..6d62333e2 100644 --- a/ml_exp/__main__.py +++ b/ml_exp/__main__.py @@ -41,11 +41,12 @@ if __name__ == '__main__': # plot_benchmarks() xyz, nc, pbe0, delta, atoms = read_qm7_data(return_atoms=True) for i in range(1): - fnm, bonds = fneig_matrix(atoms[i], xyz[i]) - am = adj_matrix(bonds) + fnm, bonds, forces = fneig_matrix(atoms[i], xyz[i], nc[i], True) + am = adj_matrix(bonds, forces) print(f'{i} first neighbor matrix\n{fnm}') print(f'{i} bond list\n{bonds}') + print(f'{i} force list\n{forces}') print(f'{i} adjacency matrix\n{am}') print('-'*30) print('OK!') |