From f5a019c0598ddd499d6bd154f3503993682959a9 Mon Sep 17 00:00:00 2001 From: Clayton O'Dell Date: Mon, 2 Dec 2024 22:49:42 -0500 Subject: [PATCH] Fix OneHotEncoder usage in sklearn example notebook Signed-off-by: Clayton O'Dell --- examples/sklearn/demo_new_features.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/sklearn/demo_new_features.ipynb b/examples/sklearn/demo_new_features.ipynb index 46b15321..fb91c52a 100644 --- a/examples/sklearn/demo_new_features.ipynb +++ b/examples/sklearn/demo_new_features.ipynb @@ -632,7 +632,7 @@ ], "source": [ "ohe = make_column_transformer(\n", - " (OneHotEncoder(sparse=False), X_train.dtypes == 'category'),\n", + " (OneHotEncoder(sparse_output=False), X_train.dtypes == 'category'),\n", " remainder='passthrough', verbose_feature_names_out=False)\n", "X_train = pd.DataFrame(ohe.fit_transform(X_train), columns=ohe.get_feature_names_out(), index=X_train.index)\n", "X_test = pd.DataFrame(ohe.transform(X_test), columns=ohe.get_feature_names_out(), index=X_test.index)\n",