From 4423524734b15fdb9ca8aafb5eefa823d70ba671 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Wed, 13 Dec 2023 18:11:54 +0800 Subject: [PATCH] fix schema --- .../tools/functions/schemas/feature_engineering.yml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/metagpt/tools/functions/schemas/feature_engineering.yml b/metagpt/tools/functions/schemas/feature_engineering.yml index 4f2a7100d..3ba9e863b 100644 --- a/metagpt/tools/functions/schemas/feature_engineering.yml +++ b/metagpt/tools/functions/schemas/feature_engineering.yml @@ -53,17 +53,17 @@ PolynomialExpansion: CatCount: type: class - description: "Add value counts of categorical columns as new features." + description: "Add value counts of a categorical column as new feature." methods: __init__: description: "Initialize self." parameters: properties: - cols: - type: list - description: "Columns for value counts." + col: + type: str + description: "Column for value counts." required: - - cols + - col fit: description: "Fit the CatCount model." parameters: