Plexe:用自然语言构建机器学习模型

通过自然语言描述意图,自动完成数据分析和模型构建的AI系统。

Star 数
★ 2.6k
最近更新
7 个月前
License
Apache-2.0
主语言
Python

30 秒速览

运行形态
命令行工具代码库 / SDK自托管服务
可在哪里用
通用 · 跨平台OpenAI API · Claude API
费用
软件免费,模型调用费用自付
上手难度
中 · 需要几步配置
开始前需要
Python 3.10-3.12OpenAI API 密钥Anthropic API 密钥Shell / 命令行网络访问本地文件系统
典型场景
数据科学家在探索新数据集时,想快速评估多种模型,只需用自然语言描述目标,就能获得一个基线和部署包。
不适合
  • 只处理图像或文本数据的团队,主要面向表格数据
  • 需要精细控制训练过程的资深 ML 从业者,自动搜索不透明

这个 Agent 能做什么,适合哪些场景?

Plexe 是一个基于多智能体架构的机器学习模型构建工具,用户只需用自然语言描述预测目标并提供表格数据(如CSV、Parquet),系统即可自动完成从数据探索、模型选择、训练到打包的全流程。它集成了14个专业智能体,覆盖6个阶段的工作流,支持XGBoost、CatBoost、LightGBM、Keras和PyTorch等主流框架。最终产出一个自包含的模型包,包含推理代码、配置和评估报告,且不依赖Plexe本身,便于部署。此外,提供Docker镜像和Streamlit可视化仪表盘,方便实验管理和结果分析。Plexe通过LiteLLM支持多种LLM提供商,并具有YAML配置和集成接口,适合需要快速验证模型想法的数据科学家和ML工程师。

Plexe 通过 python -m plexe.main 命令行或Python API接收训练数据(支持Parquet、CSV、ORC、Avro)和自然语言意图,然后启动多智能体工作流:数据分析智能体识别任务类型,假设生成智能体提出模型思路,模型定义智能体选择具体框架和超参数,评估智能体验证性能,最终生成可部署的模型包。它调用OpenAI或Anthropic的LLM API来驱动智能体,利用LiteLLM集成,支持配置不同的LLM路由。输出的模型包包含训练好的模型构件(pickle)、推理预测代码、JSON模式、配置和评估报告。用户可以定制迭代次数、允许的模型类型,并通过WorkflowIntegration接口连接自定义存储或部署系统。

  1. 数据科学家在探索新数据集时,想快速评估多种模型,只需用自然语言描述目标,就能获得一个基线和部署包。
  2. ML工程师需要为业务部门快速生成可部署的模型原型,无需手动编写训练代码,只需提供数据文件。
  3. 教育场景中,学生或研究人员想直观理解从数据到模型的完整流程,可以用Plexe演示自动建模。
  4. 使用Docker CI/CD的团队,可以集成Plexe的镜像,在自动化流水线中根据提示自动构建模型。
  5. 需要对比不同LLM提供商效果的开发者,可以配置Plexe使用OpenAI或Anthropic,测试智能体性能。
  6. 业务分析师有干净的数据集,希望不编写代码直接获得预测模型,Plexe的CLI或云端服务可降低门槛。

如何安装或部署这个 Agent?

安装Plexe需要Python 3.10至3.12,并使用pip安装:pip install plexe(核心版本包含XGBoost、Keras、scikit-learn)。可选框架支持:pip install "plexe[catboost]"、pip install "plexe[lightgbm]"、pip install "plexe[pytorch]",或按任务分组:pip install "plexe[tabular]"(CatBoost+LightGBM)、pip install "plexe[vision]"(PyTorch)。平台扩展:pip install "plexe[pyspark]"、pip install "plexe[aws]"。需要设置API密钥:export OPENAI_API_KEY=<your-key> 和 export ANTHROPIC_API_KEY=<your-key>(根据LiteLLM支持的其他提供商)。

如何使用这个 Agent?

基本用法:在命令行运行 python -m plexe.main --train-dataset-uri data.parquet --intent "predict whether a passenger was transported" --max-iterations 5。Python API:调用 from plexe.main import main,传参 intent、data_refs、max_iterations 和 work_dir。可选参数包括 allowed_model_types(如["xgboost"])和 enable_final_evaluation。配置可通过YAML文件:创建config.yaml并设置max_search_iterations、allowed_model_types、LLM路由等,然后使用CONFIG_FILE=config.yaml运行。输出在work_dir/model/目录下。Docker方式:构建镜像后运行 docker run --rm -e OPENAI_API_KEY=$OPENAI_API_KEY -e ANTHROPIC_API_KEY=$ANTHROPIC_API_KEY -v $(pwd)/data:/data -v $(pwd)/workdir:/workdir plexe:py3.12 python -m plexe.main ...。可视化:python -m plexe.viz --work-dir ./workdir 启动Streamlit仪表盘。

这个 Agent 有哪些优点和局限?

优点
  • 自然语言建模,无需手动编写训练代码,大幅降低ML入门门槛。
  • 多智能体协作,自动处理数据分析和模型选择,可支持XGBoost、CatBoost、PyTorch等框架。
  • 输出自包含模型包,不依赖Plexe,方便部署到任何环境。
  • 通过LiteLLM支持多LLM提供商,可灵活切换OpenAI/Anthropic,降低供应商锁定风险。
  • 提供Docker镜像和预配置环境,简化依赖管理。
  • 内置Streamlit仪表盘,可视化实验过程。
局限
  • 需要外部LLM API密钥,且调用LLM会产生费用,运行成本取决于迭代次数。
  • 仅支持表格数据,未提及对图像或文本数据的内置支持(但可选vision依赖可能暗示某些扩展)。
  • Python版本限制在3.10-3.12,可能妨碍使用更旧或更新的运行时。
  • 依赖多个框架,安装体积较大,且默认不包含全部框架,需手动添加extra。
  • 当前仅主动测试OpenAI和Anthropic提供商,其他LLM支持可能不稳定。
  • 构建模型的过程不透明,用户难以完全控制每个步骤,可能不适合需要精细调参的专业模型。

这个 Agent 与同类方案有什么区别?

与相关度最高的同类 agent 并排比较关键指标。

Agent 源码审查 形态 / 费用 Star 最近更新 主语言 完整支持的平台
Plexe:用自然语言构建机器学习模型 当前 41 · 缺口较多 命令行工具免费 + 模型费 ★ 2.6k 7 个月前 Python OpenAI API · Claude API
RecursiveMAS 31 · 缺口较多 命令行工具免费 + 模型费 ★ 969 20 天前 Python —
EvoScientist:自我进化的 AI 科学家 51 · 缺口较多 命令行工具免费 + 模型费 ★ 5k 2 天前 Python OpenAI API · Claude API
Principia 66 · 存在缺口 命令行工具免费 + 模型费 ★ 1.1k 17 天前 Rich Text Format —

FollowAgents 如何评估这个 Agent?

FollowAgents 源码审查 · FARS-2.1
缺口较多
41/ 100 五分制 2.1 / 5
信任安全 0/29
可靠稳定 8/14
适用触发 12/18
规范维护 11/18
有效结果 7/13
证据核验 3/8
查看各维度的扣分理由
信任安全0 / 29 · 0.0/5

证据显示:仓库未提供权限模型、用户确认机制、数据流透明性、敏感数据处理、依赖安全审计、外部影响控制、回滚机制或来源归属的说明。所有信任相关标准均未得到满足,因此得分为0。

可靠稳定8 / 14 · 2.9/5

证据显示:README和pyproject.toml在功能描述和依赖声明上保持一致,但未提供故障消息或错误处理文档。因此,自我一致性得分为2,依赖可用性得分为2,故障消息得分为1。

适用触发12 / 18 · 3.3/5

证据显示:README明确了目标受众(开发者、研究人员)和使用场景(表格数据建模),并提供了CLI和Python API的示例。能力边界通过支持模型类型和可选依赖进行了说明。触发精度通过明确的命令和参数得到保证。环境适配通过Docker和配置选项得到支持。因此,各标准得分为2。

规范维护11 / 18 · 3.1/5

证据显示:README提供了清晰的信息架构,安装说明详细,命名稳定,示例和FAQ存在,已知限制部分提及了仅测试OpenAI和Anthropic模型。许可证为Apache-2.0,版本号在pyproject.toml中,但未提供变更日志。维护责任通过贡献指南和Discord链接得到体现。因此,各标准得分为2,许可证得分为3,已知限制和版本变更日志得分为1。

有效结果7 / 13 · 2.7/5

证据显示:输出可用性通过模型包的结构和示例代码得到体现,边际价值通过自动化模型构建和多种框架支持得到体现,成本效益通过API密钥要求和云服务提及得到部分体现。因此,输出可用性和边际价值得分为2,成本效益得分为1。

证据核验3 / 8 · 1.9/5

证据显示:README中的声明部分有示例支持,但缺乏独立的验证来源。事实和推断的区分不够明确。因此,各标准得分为1。

风险与缓解建议
  • 源码中未见:最小权限约束只授予完成任务所需的权限:用专用账号或只读令牌,并限定可访问的目录和仓库。
  • 源码中未见:执行前用户确认开启或自行加上执行前确认;先在沙箱或测试环境跑通,确认行为后再接入真实数据。
  • 源码中未见:数据流向说明运行时观察它连接了哪些外部服务(代理或防火墙日志);弄清数据去向之前不要输入敏感数据。
  • 源码中未见:敏感信息处理使用专用、低权限、可随时吊销的 API 密钥,不要复用生产凭据,也不要让密钥出现在日志里。
  • 源码中未见:依赖安全审查安装前固定版本并做一次依赖扫描(如 npm audit、pip-audit);优先放在容器里运行。
  • 源码中未见:外部影响披露先弄清它会写入、发送或修改哪些外部系统,用测试账号或测试仓库验证后再接入正式环境。
  • 源码中未见:回滚或恢复路径运行前先备份,或在 git 分支、快照上操作,确保改动可以撤销。
  • 源码中未见:来源归属可核验从官方仓库或包源安装,核对发布者和仓库地址,避免同名仿冒包。
  • 未提供权限模型或用户确认机制,可能执行任意代码或访问敏感数据。
  • 依赖安全未审计,存在已知漏洞风险。
  • 未提供回滚机制,模型构建失败可能无法恢复。
  • 来源归属不明确,发布者身份未验证。
证据充分度:低 评估于 2026年8月13日 审查版本 a1e05f6dc4c0
查看完整评分方法 →

最近版本

来自该项目 GitHub Releases 的最新发布记录,说明为上游原文摘录,未经本站改写或测试。

  1. v1.4.4
    发布说明摘录
    What's Changed * chore: ignore .claude and .codex directories by @marcellodebernardi in https://github.com/plexe-ai/plexe/pull/188 * feat: support explicit split inputs and nn epoch overrides by @marcellodebernardi in https://github.com/plexe-ai/plexe/pull/189 Full Changelog: https://github.com/plexe-ai/plexe/compare/v1.4.3...v1.4.4
    在 GitHub 查看 ↗
  2. v1.4.3
    发布说明摘录
    What's Changed * feat: add probability-aware metric support across search and evaluation by @marcellodebernardi in https://github.com/plexe-ai/plexe/pull/187 Full Changelog: https://github.com/plexe-ai/plexe/compare/v1.4.2...v1.4.3
    在 GitHub 查看 ↗
  3. v1.4.2
    发布说明摘录
    What's Changed * fix: calibrate evaluator verdict rubric and baseline context by @marcellodebernardi in https://github.com/plexe-ai/plexe/pull/186 Full Changelog: https://github.com/plexe-ai/plexe/compare/v1.4.1...v1.4.2
    在 GitHub 查看 ↗

常见问题

运行Plexe需要哪些API密钥?
需要OpenAI和Anthropic的API密钥,通过环境变量设置。Plexe使用LiteLLM,你也可以配置其他支持的提供商,但只保证OpenAI和Anthropic经过测试。
Plexe支持哪些数据类型?
目前支持表格数据,格式包括Parquet、CSV、ORC和Avro。不明确支持非表格数据,如有图片需求可能需要查看vision扩展。
可以自定义模型构建过程吗?
可以。你可以在YAML配置中设置最大迭代次数、允许的模型类型、LLM路由等参数,还可以通过WorkflowIntegration接口连接自定义存储和部署系统。
模型包可以在没有Plexe的环境中使用吗?
可以。输出模型包是自包含的,包含推理代码和训练好的构件,不依赖Plexe库,可以部署到任何标准Python环境。
Plexe适合生产环境吗?
它提供模型打包和Docker支持,适合快速原型和部署准备。但对于生产级模型,你可能需要额外测试和监控,因为自动生成的模型可能不如手工调优的模型在特定用途上表现好。
在 GitHub 查看 ↗ 安装 ↓

相关 Agents