summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ml_exp/krr.py5
-rw-r--r--ml_exp/representations.py2
2 files changed, 3 insertions, 4 deletions
diff --git a/ml_exp/krr.py b/ml_exp/krr.py
index 358816a39..68aef8828 100644
--- a/ml_exp/krr.py
+++ b/ml_exp/krr.py
@@ -91,7 +91,7 @@ def krr(descriptors,
printc(f'\tTraining size: {training_size}', 'CYAN')
printc(f'\tTest size: {test_size}', 'CYAN')
printc(f'\tSigma: {sigma}', 'CYAN')
- printc(f'\tKernel: {laplauss}', 'CYAN')
+ printc(f'\tKernel: {kernel}', 'CYAN')
printc(f'\tUse tf: {use_tf}', 'CYAN')
if use_tf:
@@ -310,8 +310,7 @@ def multi_krr(db_path='data',
sort=sort,
flatten=flatten)
if 'BOB' in identifiers:
- compound.gen_bob(size=size,
- sort=sort,
+ compound.gen_bob(sort=sort,
acount=acount)
# Create a numpy array (or tensorflow tensor) for the descriptors.
diff --git a/ml_exp/representations.py b/ml_exp/representations.py
index 563b91c2f..31ab05169 100644
--- a/ml_exp/representations.py
+++ b/ml_exp/representations.py
@@ -356,7 +356,7 @@ Use non-eigenvalue representation.')
else:
bag = ''.join(sorted([atom_i, atom_j]))
- bags[bag][1].append(cm[i, j])
+ bags[bag][1].append(cm[i, j])
# Change to a numpy array and add padding.
for bag in bags.keys():