diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-26 05:32:35 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-26 05:32:35 -0700 |
| commit | a0e73407bfef3569a994c07d5bb9c8df97518e30 (patch) | |
| tree | 3232125c94fdbc3df634f8067f51965792e22c11 | |
| parent | 5a85580b5f99484b957b65326ec5a19ab627d801 (diff) | |
Add feature
| -rw-r--r-- | ml_exp/compound.py | 2 | ||||
| -rw-r--r-- | ml_exp/qm7db.py | 6 |
2 files changed, 5 insertions, 3 deletions
diff --git a/ml_exp/compound.py b/ml_exp/compound.py index 0a8b89610..9c7102860 100644 --- a/ml_exp/compound.py +++ b/ml_exp/compound.py @@ -36,6 +36,7 @@ class Compound: self.name = None self.n = None + self.extra = None # In case a comment is in the compound file. self.atoms = None self.atoms_nc = None self.coordinates = None @@ -139,6 +140,7 @@ class Compound: self.name = filename.split('/')[-1] self.n = int(lines[0]) + self.extra = lines[1] self.atoms = [] self.atoms_nc = np.empty(self.n, dtype=int) self.coordinates = np.empty((self.n, 3), dtype=float) diff --git a/ml_exp/qm7db.py b/ml_exp/qm7db.py index f9950c317..8d2453f45 100644 --- a/ml_exp/qm7db.py +++ b/ml_exp/qm7db.py @@ -49,9 +49,9 @@ def qm7db(nc, compounds[i].pbe0 = float(line[1]) compounds[i].delta = float(line[1]) - float(line[2]) - # Shuffle the compounds list - random.seed(r_seed) - random.shuffle(compounds) + if is_shuffled: + random.seed(r_seed) + random.shuffle(compounds) e_pbe0 = np.array([compound.pbe0 for compound in compounds], dtype=float) e_delta = np.array([compound.delta for compound in compounds], dtype=float) |