AIDE: AI-Driven Exploration in the Space of Code. The machine Learning engineering agent that automates AI R&D.
AIDE ML 是一个基于大语言模型的开源智能体框架,专门用于机器学习领域的自动化研发。它通过构建代理式树搜索算法,能够自动完成机器学习代码的编写、调试、评估及优化,旨在最大化或最小化用户指定的目标指标。
支持通过自然语言描述任务,仅需指定数据集、目标和评估指标即可启动自动化流程。 采用代理式树搜索算法,将生成的 Python 脚本作为解空间节点,通过迭代反馈不断演进代码方案。 内置包括 CLI 命令行工具、可视化界面和配置预设在内的完整工具链,方便用户快速上手。 在标准化基准测试中表现卓越,能够通过自我改进和多路径搜索产生比传统线性智能体更优的建模方案。
该项目适用于研究代理架构的科研人员,以便于测试搜索启发式策略或集成新的大模型后端;同时也非常适合机器学习从业者,帮助他们基于数据集快速构建和调优高性能的预测管道。