diff options
author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-21 18:13:33 -0700 |
---|---|---|
committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-21 18:13:33 -0700 |
commit | efaa8eeb58d8e19d32f94ec4e91ec794574dfd32 (patch) | |
tree | 02c5250412880b0d50aebd028e975902dd809c18 /ml_exp/representations.py | |
parent | bb5e608f0ca6491e1abfd7fd25d2f28ff6c96e6b (diff) |
Coulomb matrix done, fix init
Diffstat (limited to 'ml_exp/representations.py')
-rw-r--r-- | ml_exp/representations.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/ml_exp/representations.py b/ml_exp/representations.py index bbc538ae6..c503c310d 100644 --- a/ml_exp/representations.py +++ b/ml_exp/representations.py @@ -28,16 +28,16 @@ def coulomb_matrix(coords, nc, size=23, as_eig=True, - bohr_radius_units=False): + bhor_ru=False): """ Creates the Coulomb Matrix from the molecule data given. coords: compound coordinates. nc: nuclear charge data. size: compound size. - as_eig: if data should be returned as matrix or array of eigenvalues. - bohr_radius_units: if units should be in bohr's radius units. + as_eig: if the representation should be as the eigenvalues. + bhor_ru: if radius units should be in bohr's radius units. """ - if bohr_radius_units: + if bhor_ru: cr = 0.52917721067 else: cr = 1 @@ -54,7 +54,7 @@ def coulomb_matrix(coords, size = n nr = range(size) - cm = np.empty((size, size), dtype=float) + cm = np.zeros((size, size), dtype=float) # Actual calculation of the coulomb matrix. for i in nr: |