1 files changed, 4 insertions, 1 deletions
diff --git a/do_ml.py b/do_ml.py
index 87e8b72db..a424a610a 100644
--- a/do_ml.py
+++ b/do_ml.py
@@ -68,7 +68,10 @@ def do_ml(desc_data,
         return None
 
     if not test_size:
-        test_size = d_len - training_size
+        if d_len - training_size > 1500:
+            test_size = 1500
+        else:
+            test_size = d_len - training_size
 
     tic = time.perf_counter()
     if show_msgs: