Modified version conflict about scikit-learn.

DataCanvasIO · Feb 23, 2024 · f095f19 · f095f19
1 parent 6e4c2ff
commit f095f19
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/hyperts/utils/transformers.py b/hyperts/utils/transformers.py
@@ -299,7 +299,12 @@ def __init__(self, label_encoder=None, onehot_encoder=None, copy=True):
         else:
             self.label_encoder = label_encoder
         if onehot_encoder is None:
-            self.onehot_encoder = OneHotEncoder(sparse=False, categories="auto")
+            if hasattr(OneHotEncoder(), "sparse"):
+                self.onehot_encoder = OneHotEncoder(sparse=False, categories="auto")
+            elif hasattr(OneHotEncoder(), "sparse_output"):
+                self.onehot_encoder = OneHotEncoder(sparse_output=False, categories="auto")
+            else:
+                raise ValueError("please check the version of scikit-learn package.")
         else:
             self.onehot_encoder = onehot_encoder