diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-21 19:18:24 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-21 19:18:24 -0700 |
| commit | 05d97a15d822400f7018a4381593c2638c6ad577 (patch) | |
| tree | f39be8d83d1724d6005e5e8f63c87af5de1e1a4e | |
| parent | 5ba008a928d34c49e431f32b5161f624d780c254 (diff) | |
Refactor code
| -rw-r--r-- | ml_exp/compound.py | 44 |
1 files changed, 22 insertions, 22 deletions
diff --git a/ml_exp/compound.py b/ml_exp/compound.py index d74f1230a..c6d5f8a64 100644 --- a/ml_exp/compound.py +++ b/ml_exp/compound.py @@ -40,8 +40,8 @@ class Compound: self.coordinates = None self.energy = None - self.c_matrix = None - self.lj_matrix = None + self.cm = None + self.ljm = None self.bob = None if xyz is not None: @@ -57,19 +57,19 @@ class Compound: as_eig: if the representation should be as the eigenvalues. bhor_ru: if radius units should be in bohr's radius units. """ - self.c_matrix = coulomb_matrix(self.coordinates, - self.atoms_nc, - size=size, - as_eig=as_eig, - bhor_ru=bohr_ru) + self.cm = coulomb_matrix(self.coordinates, + self.atoms_nc, + size=size, + as_eig=as_eig, + bhor_ru=bohr_ru) - def gen_lj(self, - diag_value=None, - sigma=1.0, - epsilon=1.0, - size=23, - as_eig=True, - bohr_ru=False): + def gen_ljm(self, + diag_value=None, + sigma=1.0, + epsilon=1.0, + size=23, + as_eig=True, + bohr_ru=False): """ Generate the Lennard-Jones Matrix for the compund. diag_value: if special diagonal value is to be used. @@ -79,14 +79,14 @@ class Compound: as_eig: if the representation should be as the eigenvalues. bhor_ru: if radius units should be in bohr's radius units. """ - self.lj_matrix = lennard_jones_matrix(self.coordinates, - self.atoms_nc, - diag_value=diag_value, - sigma=sigma, - epsilon=epsilon, - size=size, - as_eig=as_eig, - bhor_ru=bohr_ru) + self.ljm = lennard_jones_matrix(self.coordinates, + self.atoms_nc, + diag_value=diag_value, + sigma=sigma, + epsilon=epsilon, + size=size, + as_eig=as_eig, + bhor_ru=bohr_ru) def read_xyz(self, filename): |