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

add ml example

Browse source
This commit is contained in:
yzlin 2024-02-08 17:09:07 +08:00
parent c666f0e24e
commit 74c5807d01
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)