This commit is contained in:
lidanyang 2023-12-12 13:39:41 +08:00
parent 988c7072ef
commit 9c426f73dc

View file

@ -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