updated a few more calls to OneHotEncoder

mindsdb · Dec 4, 2024 · 404eb1c · 404eb1c
1 parent db8038e
commit 404eb1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightwood/analysis/nc/calibrate.py b/lightwood/analysis/nc/calibrate.py
@@ -59,7 +59,7 @@ def analyze(self, info: Dict[str, object], **kwargs) -> Dict[str, object]:
                 all_classes = np.array([ns.encoded_val_data.encoders[ns.target].rev_map[idx] for idx in class_keys])
 
             if data_type != dtype.tags:
-                enc = OneHotEncoder(sparse=False, handle_unknown='ignore')
+                enc = OneHotEncoder(sparse_output=False, handle_unknown='ignore')
                 enc.fit(all_classes.reshape(-1, 1))
                 output['label_encoders'] = enc  # needed to repr cat labels inside nonconformist
             else: