基于 LangGraph 的多智能体研究助手,可生成假设、进行数据分析与可视化并撰写研究报告。
WecoAI/aideml
开源 LLM 智能体,通过树搜索自动编写、调试并迭代机器学习代码,直到优化用户指定的指标。
- Star
- 1,614
- Fork
- 237
- 未关闭 Issue
- 3
- 最近提交
- 22天前
这是什么?
AIDE ML 是 AIDE 算法的开源参考实现。AIDE 是一种树搜索式智能体,把每个 Python 脚本当作解决方案树中的一个节点。用户提供数据目录、用自然语言描述的目标以及评估指标后,它会生成代码、执行,并依据指标反馈决定扩展哪些候选方案、修复哪些错误。
项目以 Python 包形式发布,可通过 pip 安装,提供命令行工具、解决方案树的 HTML 可视化以及 Streamlit 网页界面。它支持 OpenAI、Anthropic、Gemini、OpenRouter,也支持任何兼容 OpenAI API 的本地模型。配置项包括代码模型、反馈模型、迭代步数和草稿数量,并提供 Python 接口与 Docker 镜像。
该项目适合希望复现 AIDE 论文或尝试新搜索策略、评估器与 LLM 后端的研究人员,也适合希望基于数据集快速搭建机器学习流程的工程师。
核心功能
- 01用自然语言描述任务,并提供数据目录、目标与评估指标即可运行。
- 02在生成的 Python 脚本树上搜索,依据指标反馈扩展或修复候选方案。
- 03提供命令行工具、Python 接口和 Docker 镜像。
- 04包含解决方案树的 HTML 可视化与 Streamlit 网页界面。
- 05支持 OpenAI、Anthropic、Gemini、OpenRouter 以及兼容 OpenAI 接口的本地模型。
适用场景
- 根据自然语言目标,在表格数据集上快速搭建机器学习基线模型。
- 在研究中复现或扩展 AIDE 的树搜索实验。
- 测试机器学习智能体的不同搜索策略、评估器或 LLM 后端。
- 结合 Ollama 等本地模型,在本地完成全流程运行。
快速上手
- 1
从 PyPI 安装软件包
pip install -U aideml - 2
设置 LLM API 密钥
export OPENAI_API_KEY=<your-key> - 3
在示例任务上运行优化
aide data_dir="example_tasks/house_prices" goal="Predict the sales price for each house" eval="RMSE between log-prices"
使用前须知
- 默认设置会调用外部 LLM API,因此需要 API 密钥,并且除非使用本地模型,否则会产生模型使用费用。
- 完全本地运行需要兼容 OpenAI 接口的本地服务,通常与托管模型相比性能会有所下降。
收录于专题
标签
常见问题
- aideml 是什么?
- 开源 LLM 智能体,通过树搜索自动编写、调试并迭代机器学习代码,直到优化用户指定的指标。 AIDE ML 是 AIDE 算法的开源参考实现。AIDE 是一种树搜索式智能体,把每个 Python 脚本当作解决方案树中的一个节点。用户提供数据目录、用自然语言描述的目标以及评估指标后,它会生成代码、执行,并依据指标反馈决定扩展哪些候选方案、修复哪些错误。
- aideml 可以免费使用吗?用的什么许可证?
- aideml 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
- aideml 还在维护吗?
- 活跃维护中。aideml 最近一次提交是在 2026年10月16日。
- aideml 适合用来做什么?
- 根据自然语言目标,在表格数据集上快速搭建机器学习基线模型。 在研究中复现或扩展 AIDE 的树搜索实验。 测试机器学习智能体的不同搜索策略、评估器或 LLM 后端。 结合 Ollama 等本地模型,在本地完成全流程运行。
- aideml 怎么安装和运行?
- 从 PyPI 安装软件包: pip install -U aideml 设置 LLM API 密钥: export OPENAI_API_KEY=<your-key> 在示例任务上运行优化: aide data_dir="example_tasks/house_prices" goal="Predict the sales price for each house" eval="RMSE between log-prices"
- 在哪里获取 aideml?
- 源代码和安装说明在 GitHub 上:https://github.com/WecoAI/aideml
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
一个将训练好的 scikit-learn 模型和转换器转译为 C、Go、Java、JavaScript 等语言代码的 Python 工具。
基于 PyTorch 的深度学习框架,自动处理训练循环、多 GPU 扩展等工程代码,同时保留模型逻辑的完全控制。
微软开源的 LLM 智能体框架,自动化数据驱动的研发流程,包括论文复现、模型与因子实现和 Kaggle 式实验。
一个递归自我改进的 Agent 运行框架,通过评估迭代任务,保留有效经验并为下一轮留下轨迹与操作手册。
AutoML 库,用几行 Python 代码即可对表格、时间序列和多模态数据自动训练并集成模型。