another onehotencoder keyword update

ai2es · Feb 21, 2024 · f0e00a0 · f0e00a0
1 parent e22fcd3
commit f0e00a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ptype/data.py b/ptype/data.py
@@ -327,7 +327,7 @@ def preprocess_data(
         )
 
     if encoder_type == "onehot":
-        scalers["output_onehot"] = OneHotEncoder(sparse=False)
+        scalers["output_onehot"] = OneHotEncoder(sparse_output=False)
         scaled_data["train_y"] = scalers["output_onehot"].fit_transform(
             np.expand_dims(scaled_data["train_y"], 1)
         )