From 9c426f73dc7ba876fbe076a4d5f71996424fcfcf Mon Sep 17 00:00:00 2001 From: lidanyang Date: Tue, 12 Dec 2023 13:39:41 +0800 Subject: [PATCH] fix bug --- metagpt/tools/functions/libs/feature_engineering.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/tools/functions/libs/feature_engineering.py b/metagpt/tools/functions/libs/feature_engineering.py index 06a988d9a..67247d0d1 100644 --- a/metagpt/tools/functions/libs/feature_engineering.py +++ b/metagpt/tools/functions/libs/feature_engineering.py @@ -283,5 +283,5 @@ class GeneralSelection(MLProcess): self.feats = feats def transform(self, df: pd.DataFrame) -> pd.DataFrame: - df = df[self.feats] + df = df[self.feats + [self.label_col]] return df