[MRG] improve the task, model planner for csv examples - Githubissues

MLSysOps / MLE-agent

MLE-Agent is designed to be a pair coding agent for machine learning engineers and researchers. Support OpenAI and Ollama.

MIT License

17 stars 1 forks source link

[MRG] improve the task, model planner for csv examples #34

Closed HuaizhengZhang closed 1 month ago

HuaizhengZhang commented 1 month ago

Closes

33
[x] improve mle start logic
[x] improve task/model selection prompt

What has been done to verify that this works as intended?

Why is this the best possible solution? Were any other approaches considered?

How does this change affect users? Describe intentional changes to behavior and behavior that could have accidentally been affected by code changes. In other words, what are the regression risks?

Do we need any specific form for testing your changes? If so, please attach one.

Does this change require updates to documentation? If so, please file an issue here and include the link below.

Before submitting this PR, please make sure you have:

[x] confirmed all checks still pass OR confirm CI build passes.
[x] verified that any code or assets from external sources are properly credited in comments and/or in the credit file.

HuaizhengZhang commented 1 month ago

now even with "use csv to train a model" as a requirement, we can give a right task, model and plan.

compared to

33