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

Merge pull request #886 from garylin2099/ci_dev

Browse source 
add ml example
This commit is contained in:
garylin2099 2024-02-08 17:25:03 +08:00 • committed by GitHub
commit 2d9b1510d3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
examples/ci/machine_learning.py
View file

@ -0,0 +1,13 @@
import asyncio
from metagpt.roles.ci.code_interpreter import CodeInterpreter
async def main(requirement: str):
role = CodeInterpreter(auto_run=True, use_tools=False)
await role.run(requirement)
if __name__ == "__main__":
requirement = "Run data analysis on sklearn Wine recognition dataset, include a plot, and train a model to predict wine class (20% as validation), and show validation accuracy."
asyncio.run(main(requirement))

4
examples/ci/ml_engineer_with_tools.py
View file

@ -3,8 +3,8 @@ import asyncio
from metagpt.roles.ci.ml_engineer import MLEngineer
async def main(requirement: str, auto_run: bool = True, use_tools: bool = True):
role = MLEngineer(goal=requirement, auto_run=auto_run, use_tools=use_tools)
async def main(requirement: str):
role = MLEngineer(auto_run=True, use_tools=True)
await role.run(requirement)