From 735c6128183f3a6acb3ffee5c0387275aed6d14c Mon Sep 17 00:00:00 2001 From: lidanyang Date: Tue, 19 Mar 2024 11:15:24 +0800 Subject: [PATCH] fix bug of old version param --- metagpt/tools/libs/data_preprocess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/tools/libs/data_preprocess.py b/metagpt/tools/libs/data_preprocess.py index aa9070689..ef02d8f3a 100644 --- a/metagpt/tools/libs/data_preprocess.py +++ b/metagpt/tools/libs/data_preprocess.py @@ -171,7 +171,7 @@ class OneHotEncode(DataPreprocessTool): def __init__(self, features: list): self.features = features - self.model = OneHotEncoder(handle_unknown="ignore", sparse=False) + self.model = OneHotEncoder(handle_unknown="ignore", sparse_output=False) def transform(self, df: pd.DataFrame) -> pd.DataFrame: ts_data = self.model.transform(df[self.features])