Update sparse to sparse_output

sklearn will deprecate sparse in favor of sparse_output, currently throwing warnings
MLBazaar · Jul 14, 2023 · ab32286 · ab32286
1 parent c95fef3
commit ab32286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/btb/tuning/hyperparams/categorical.py b/btb/tuning/hyperparams/categorical.py
@@ -49,7 +49,7 @@ def __init__(self, choices, default=NO_DEFAULT):
         self.dimensions = len(choices)
         self.cardinality = self.dimensions
         choices = np.array(choices, dtype='object')
-        self._encoder = OneHotEncoder(categories=[choices], sparse=False)
+        self._encoder = OneHotEncoder(categories=[choices], sparse_output=False)
         self._encoder.fit(choices.reshape(-1, 1))
 
     def _within_hyperparam_space(self, values):