Merge pull request #38 from ai2es/gantos

updating sklearn.onehotencoder keyword from sparse to sparse_output
ai2es · Feb 21, 2024 · e22fcd3 · e22fcd3
2 parents c49b2db + 0c30e8c
commit e22fcd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ptype/data.py b/ptype/data.py
@@ -285,7 +285,7 @@ def preprocess_data(
     scalers["input"] = scalar_obs[scaler_type]
     scalers["output_label"] = LabelEncoder()
     if encoder_type == "onehot":
-        scalers["output_onehot"] = OneHotEncoder(sparse=False)
+        scalers["output_onehot"] = OneHotEncoder(sparse_output=False)
 
     if groupby and "quantile" not in scaler_type:
         scaled_data["train_x"] = pd.DataFrame(