diff options
author | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-12 19:20:24 -0700 |
---|---|---|
committer | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-12 19:20:24 -0700 |
commit | 5ac02bfda4b3ff0700344a1b8757ec3d586ea7a5 (patch) | |
tree | e8c6b3b5e427a1f52eaea8e803c556240726be96 /main.py | |
parent | 79315ef03abeb43dfb5c4cb1b68e1a33349f2bfb (diff) |
Reformat test size
Diffstat (limited to 'main.py')
-rw-r--r-- | main.py | 14 |
1 files changed, 12 insertions, 2 deletions
@@ -54,8 +54,18 @@ cm_data = c_matrix_multiple(molecules, nuclear_charge, as_eig=True) ljm_data = lj_matrix_multiple(molecules, nuclear_charge, as_eig=True) # ML calculation. -do_ml(cm_data, energy_pbe0, 1000, 100, sigma=1000.0, desc_type='CM') -do_ml(ljm_data, energy_pbe0, 1000, 100, sigma=1000.0, desc_type='L-JM') +do_ml(cm_data, + energy_pbe0, + 1000, + test_size=100, + sigma=1000.0, + desc_type='CM') +do_ml(ljm_data, + energy_pbe0, + 1000, + test_size=100, + sigma=1000.0, + desc_type='L-JM') # End of program end_time = time.perf_counter() |