Fix OneHotEncoder sparse parameter

BirkhoffG · Apr 3, 2024 · 685ab35 · 685ab35
1 parent e38b222
commit 685ab35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/legacy/00_utils.ipynb b/nbs/legacy/00_utils.ipynb
@@ -288,7 +288,7 @@
    "outputs": [],
    "source": [
     "cat_idx = 2\n",
-    "ohe = OneHotEncoder(sparse=False)\n",
+    "ohe = OneHotEncoder(sparse_output=False)\n",
     "x_cat = ohe.fit_transform(x[:, cat_idx:])\n",
     "x_cont = x[:, :cat_idx].astype(float)\n",
     "x_transformed = np.concatenate(\n",