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

minimize ml_engineer

Browse source
This commit is contained in:
yzlin 2024-01-16 17:54:38 +08:00
parent 8a14dde219
commit c8858cd8d4
6 changed files with 51 additions and 100 deletions
Download patch file Download diff file Expand all files Collapse all files

1
metagpt/tools/tool_data_type.py
View file

@ -4,6 +4,7 @@ from pydantic import BaseModel
class ToolTypeEnum(Enum):
EDA = "eda"
DATA_PREPROCESS = "data_preprocess"
FEATURE_ENGINEERING = "feature_engineering"
MODEL_TRAIN = "model_train"

6
metagpt/tools/tool_types.py
View file

@ -8,6 +8,12 @@ from metagpt.tools.tool_data_type import ToolType, ToolTypeEnum
from metagpt.tools.tool_registry import register_tool_type
@register_tool_type
class EDA(ToolType):
name: str = ToolTypeEnum.EDA.value
desc: str = "Useful for performing exploratory data analysis"
@register_tool_type
class DataPreprocess(ToolType):
name: str = ToolTypeEnum.DATA_PREPROCESS.value