refine schema

This commit is contained in:
lidanyang 2023-12-06 14:17:29 +08:00
parent 962632cd15
commit 6edbed8fb6

View file

@ -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.")