From cdbb1ac890cb0d062cdb2f216c347f681fbfa7b8 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Sat, 28 Dec 2019 10:47:20 -0700
Subject: Fix bug

---
 lj_matrix/do_ml.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'lj_matrix/do_ml.py')

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()
-- 
cgit v1.2.3-54-g00ecf