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])