diff options
| -rw-r--r-- | ml_exp/krr.py | 5 | ||||
| -rw-r--r-- | ml_exp/representations.py | 2 |
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(): |