diff options
author | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-28 10:47:20 -0700 |
---|---|---|
committer | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-28 10:47:20 -0700 |
commit | cdbb1ac890cb0d062cdb2f216c347f681fbfa7b8 (patch) | |
tree | e7c0bf766e59f8963e994552af4df05522ed8f70 /lj_matrix/do_ml.py | |
parent | b4c2dc01ab17248814988c8e141bf16072c45abd (diff) |
Fix bug
Diffstat (limited to 'lj_matrix/do_ml.py')
-rw-r--r-- | lj_matrix/do_ml.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/lj_matrix/do_ml.py b/lj_matrix/do_ml.py index 12323780a..8724e6831 100644 --- a/lj_matrix/do_ml.py +++ b/lj_matrix/do_ml.py @@ -113,7 +113,7 @@ def ml(desc_data, def do_ml(min_training_size, max_training_size=None, - training_increment_size=None, + training_increment_size=500, ljm_sigma=1.0, ljm_epsilon=1.0, save_benchmarks=False, @@ -138,6 +138,8 @@ def do_ml(min_training_size, """ # Initialization time. init_time = time.perf_counter() + if not max_training_size: + max_training_size = min_training_size + training_increment_size # Data reading. molecules, nuclear_charge, energy_pbe0, energy_delta = read_qm7_data() |