From 6edbed8fb6e9ea19c2fa37de8d7f74888b83b903 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Wed, 6 Dec 2023 14:17:29 +0800 Subject: [PATCH] refine schema --- metagpt/tools/functions/schemas/feature_engineering.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/tools/functions/schemas/feature_engineering.py b/metagpt/tools/functions/schemas/feature_engineering.py index c14bb933e..df2eebff6 100644 --- a/metagpt/tools/functions/schemas/feature_engineering.py +++ b/metagpt/tools/functions/schemas/feature_engineering.py @@ -20,7 +20,7 @@ class PolynomialExpansion(ToolSchema): class OneHotEncoding(ToolSchema): - """Apply one-hot encoding to specified categorical columns in a DataFrame.""" + """Apply one-hot encoding to specified categorical columns, the original columns will be dropped.""" df: pd.DataFrame = tool_field(description="DataFrame to process.") cols: list = tool_field(description="Categorical columns to be one-hot encoded.")