From 56b64719ebec9c24d4c421f0b059640308e49ed8 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Thu, 12 Dec 2019 23:38:35 -0700
Subject: Fix bug
---
do_ml.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/do_ml.py b/do_ml.py
index a424a610a..c88533e68 100644
--- a/do_ml.py
+++ b/do_ml.py
@@ -68,10 +68,9 @@ def do_ml(desc_data,
return None
if not test_size:
- if d_len - training_size > 1500:
+ test_size = d_len - training_size
+ if test_size > 1500:
test_size = 1500
- else:
- test_size = d_len - training_size
tic = time.perf_counter()
if show_msgs:
--
cgit v1.2.3-70-g09d2