From db8038ea98c0d3af707fbf59058e425c798ea5a2 Mon Sep 17 00:00:00 2001 From: Minura Punchihewa Date: Wed, 4 Dec 2024 21:48:09 +0530 Subject: [PATCH] updated use of OneHotEncoder with bumped version --- lightwood/encoder/helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightwood/encoder/helpers.py b/lightwood/encoder/helpers.py index ff3359a88..b3fceb173 100644 --- a/lightwood/encoder/helpers.py +++ b/lightwood/encoder/helpers.py @@ -45,7 +45,7 @@ class CatNormalizer: def __init__(self, encoder_class='one_hot'): self.encoder_class = encoder_class if encoder_class == 'one_hot': - self.scaler = OneHotEncoder(sparse=False, handle_unknown='ignore') + self.scaler = OneHotEncoder(sparse_output=False, handle_unknown='ignore') else: self.scaler = OrdinalEncoder()