diff options
Diffstat (limited to 'do_ml.py')
-rw-r--r-- | do_ml.py | 5 |
1 files changed, 4 insertions, 1 deletions
@@ -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: |