apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

rm redundant docstring

Browse source
This commit is contained in:
yzlin 2024-02-04 23:27:04 +08:00
parent 4cfee3ba27
commit 8c65ed02b8
1 changed files with 0 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

15
metagpt/tools/libs/data_preprocess.py
View file

@ -71,26 +71,11 @@ class DataPreprocessTool(MLProcess):
self.model = None # to be filled by specific subclass Tool
def fit(self, df: pd.DataFrame):
"""
Fit a model to be used in subsequent transform.
Args:
df (pd.DataFrame): The input DataFrame.
"""
if len(self.features) == 0:
return
self.model.fit(df[self.features])
def transform(self, df: pd.DataFrame) -> pd.DataFrame:
"""
Transform the input DataFrame with the fitted model.
Args:
df (pd.DataFrame): The input DataFrame.
Returns:
pd.DataFrame: The transformed DataFrame.
"""
if len(self.features) == 0:
return df
new_df = df.copy()