Add shuffle import to train

Teichlab · Jun 22, 2021 · 935cbe9 · 935cbe9
1 parent 715cd98
commit 935cbe9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/celltypist/train.py b/celltypist/train.py
@@ -9,6 +9,7 @@
 from . import logger
 from scipy.sparse import spmatrix
 from datetime import datetime
+from sklearn.utils import shuffle
 
 def _to_vector(_vector_or_file):
     """