diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-26 20:22:42 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-26 20:22:42 -0700 |
| commit | a2d5786e19838788ad0d7305ec2bcfc4972d3e64 (patch) | |
| tree | dda17bc7887defe5e4ac4d1cbd7bf27c5af97e1f | |
| parent | 1113483da7e054541c7ea17ca40b641ad8e24d88 (diff) | |
Fix bob
| -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(): |