automatic detection of multioutput datasets

EpistasisLab · Jan 14, 2020 · 3dd6475 · 3dd6475
1 parent 410d88c
commit 3dd6475
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tpot/base.py b/tpot/base.py
@@ -1158,7 +1158,7 @@ def _check_dataset(self, features, target, sample_weight=None):
 
         try:
             if target is not None:
-                X, y = check_X_y(features, target, accept_sparse=True, dtype=None)
+                X, y = check_X_y(features, target, accept_sparse=True, dtype=None, multi_output=len(target.shape) > 1 and target.shape[1] > 1)
                 if self._imputed:
                     return X, y
                 else: