From 49779d8615e4b05b759b549b6d7ceb9b5258ec0a Mon Sep 17 00:00:00 2001
From: lidanyang <lidanyang@fuzhi.ai>
Date: Wed, 13 Dec 2023 13:35:22 +0800
Subject: [PATCH] refine schema desc

---
 metagpt/tools/functions/schemas/feature_engineering.yml | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/metagpt/tools/functions/schemas/feature_engineering.yml b/metagpt/tools/functions/schemas/feature_engineering.yml
index 2cc4ec2fa..4f2a7100d 100644
--- a/metagpt/tools/functions/schemas/feature_engineering.yml
+++ b/metagpt/tools/functions/schemas/feature_engineering.yml
@@ -328,7 +328,7 @@ GroupStat:
 
 SplitBins:
   type: class
-  description: "Bin continuous data into intervals and return the bin identifier encoded as an integer value"
+  description: "Inplace binning of continuous data into intervals, returning integer-encoded bin identifiers directly."
   methods:
     __init__:
       description: "Initialize self."
@@ -336,11 +336,15 @@ SplitBins:
         properties:
           cols:
             type: list
-            description: "Columns to be binned."
+            description: "Columns to be binned inplace."
           strategy:
             type: str
             description: "Strategy used to define the widths of the bins."
             default: quantile
+            enum:
+              - quantile
+              - uniform
+              - kmeans
         required:
           - cols
     fit: