From 36a50c3cbb6920b055962e10d75e8fd0939e7f82 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Thu, 12 Dec 2019 23:37:02 -0700
Subject: Setup for new benchmark and bugfix

---
 do_ml.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'do_ml.py')

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