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

test auto mode

Browse source
This commit is contained in:
stellahsr 2023-12-13 12:56:54 +08:00
parent 32c4a55755
commit 2e4094c7a7
2 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -166,4 +166,3 @@ metagpt/roles/idea_agent.py
.aider*
/config/config.yaml
/tests/metagpt/actions/check_data.py
/config/config.yaml

2
metagpt/roles/ml_engineer.py
View file

@ -223,7 +223,7 @@ if __name__ == "__main__":
# requirement = "Run data analysis on sklearn Wisconsin Breast Cancer dataset, include a plot, train a model to predict targets (20% as validation), and show validation accuracy"
requirement = "Run EDA and visualization on this dataset, train a model to predict survival, report metrics on validation set (20%), dataset: workspace/titanic/train.csv"
async def main(requirement: str = requirement, auto_run: bool = False):
async def main(requirement: str = requirement, auto_run: bool = True):
role = MLEngineer(goal=requirement, auto_run=auto_run)
await role.run(requirement)