summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ml_exp/bob.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/ml_exp/bob.py b/ml_exp/bob.py
index e3d54c5fa..20085d957 100644
--- a/ml_exp/bob.py
+++ b/ml_exp/bob.py
@@ -49,7 +49,8 @@ def bob(c_matrix,
Creates the bag of bond using the coulomb matrix data.
c_matrix: coulomb matrix.
atoms: list of atoms.
- max_len: maximum amount of bonds in molecule.
+ max_n: maximum amount of atoms.
+ max_bond_len: maximum amount of bonds in molecule.
"""
n = len(atoms)
bond_n = (n * n - n) / 2 + n