summaryrefslogtreecommitdiff
path: root/ml_exp/representations.py
diff options
context:
space:
mode:
authorDavid Luevano Alvarado <55825613+luevano@users.noreply.github.com>2020-02-29 14:07:24 -0700
committerDavid Luevano Alvarado <55825613+luevano@users.noreply.github.com>2020-02-29 14:07:24 -0700
commit4a3f602aab73bb0304a237ddca5d9f3fa8867001 (patch)
tree3039bd062cb567011afc7f2a7796edeb28eb3c58 /ml_exp/representations.py
parent53c84552df5c7a7e1f084b0aec8eac63f7813e52 (diff)
Fix bugs
Diffstat (limited to 'ml_exp/representations.py')
-rw-r--r--ml_exp/representations.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/ml_exp/representations.py b/ml_exp/representations.py
index 31259c79c..6b5d26bcb 100644
--- a/ml_exp/representations.py
+++ b/ml_exp/representations.py
@@ -358,10 +358,12 @@ generated as the vector of eigenvalues, try (re-)generating the CM.')
for j, num in enumerate(sorted(bags[checker[1]][1:])[::-1]):
# Use c_i as the index for bob if the zero padding should
# be at the end of the vector instead of between each bond.
- bob[i*size + j] = num
+ # bob[i*size + j] = num
+ bob[c_i] = num
c_i += 1
else:
print(f'Error. Bond {bond} from bond list coudn\'t be found',
'in the bags list. This could be a case where the atom',
'is only present oncce in the molecule.')
+
return bob