mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-05-24 14:15:17 +02:00
fix bug
This commit is contained in:
parent
988c7072ef
commit
9c426f73dc
1 changed files with 1 additions and 1 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue