diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-29 07:46:27 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-29 07:46:27 -0700 |
| commit | 759b48a965470ac18910951b2cb0599d371d1808 (patch) | |
| tree | 52d56f786aa112112269840f4872fe23e5118e2c | |
| parent | e289a2de7a461a1194a47350e5339dce082972e7 (diff) | |
bos to bob
| -rw-r--r-- | ml_exp/do_ml.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/ml_exp/do_ml.py b/ml_exp/do_ml.py index d8ee415bf..3cb67675f 100644 --- a/ml_exp/do_ml.py +++ b/ml_exp/do_ml.py @@ -168,9 +168,8 @@ def do_ml(db_path='data', compound.gen_am(use_forces=use_forces, size=size, bohr_ru=bohr_ru) - if 'BOS' in identifiers: - compound.gen_bos(size=size, - stuff=stuff) + if 'BOB' in identifiers: + compound.gen_bob(size=size) # Create a numpy array for the descriptors. if 'CM' in identifiers: @@ -179,8 +178,8 @@ def do_ml(db_path='data', ljm_data = np.array([comp.ljm for comp in compounds], dtype=float) if 'AM' in identifiers: am_data = np.array([comp.cm for comp in compounds], dtype=float) - if 'BOS' in identifiers: - bos_data = np.array([comp.bos for comp in compounds], dtype=float) + if 'BOB' in identifiers: + bob_data = np.array([comp.bob for comp in compounds], dtype=float) toc = time.perf_counter() tictoc = toc - tic @@ -212,8 +211,8 @@ def do_ml(db_path='data', sigma=sigma, identifier='CM', show_msgs=show_msgs) - if 'BOS' in identifiers: - bos_mae, bos_tictoc = simple_ml(bos_data, + if 'BOB' in identifiers: + bob_mae, bob_tictoc = simple_ml(bob_data, energy_pbe0, training_size=training_size, test_size=test_size, |