Repolode
plexe-ai

plexe-ai/plexe

通过自然语言描述需求和表格数据,由多智能体流程自动构建机器学习模型。

Star
2,642
Fork
263
未关闭 Issue
18
最近提交
8个月前

这是什么?

plexe 允许用自然语言描述目标来创建机器学习模型。用户提供表格数据(支持 Parquet、CSV、ORC、Avro)和一句任务描述,例如“预测乘客是否被传送”,系统中的 AI 智能体会自动搜索并训练出可用模型。

整个流程由 14 个专门的智能体分为 6 个阶段完成,包括数据分析、识别机器学习任务、选择评估指标、迭代候选模型、评估性能和打包结果。表格数据支持 XGBoost、CatBoost、LightGBM、Keras 和 PyTorch。大模型调用通过 LiteLLM 路由,不同智能体可以使用不同服务商,但项目目前主要测试的是 OpenAI 和 Anthropic 模型。

输出是一个自包含的模型包,包含训练产物、推理代码、输入输出 schema 和评估报告,部署时不依赖 plexe。项目提供带 PySpark 和 Java 的 Docker 镜像,并附带 Streamlit 面板用于查看实验结果。

核心功能

  • 01根据自然语言需求和表格数据自动构建模型。
  • 02采用多智能体、基于假设迭代的方式搜索候选模型。
  • 03表格数据支持 XGBoost、CatBoost、LightGBM、Keras 和 PyTorch。
  • 04通过 LiteLLM 和 YAML 配置为不同智能体指定大模型服务商。
  • 05导出不依赖 plexe 的自包含模型包,便于独立部署。
  • 06提供内置 PySpark 的 Docker 镜像和 Streamlit 实验面板。

适用场景

  • 在表格数据上快速原型化预测模型,无需手写训练代码。
  • 生成基线模型,用于与人工构建的流程对比。
  • 将训练好的模型打包,部署到独立的推理环境中。

快速上手

  1. 1

    从 PyPI 安装软件包

    pip install plexe
  2. 2

    设置 LLM 提供商的 API 密钥

    export OPENAI_API_KEY=<your-key> export ANTHROPIC_API_KEY=<your-key>
  3. 3

    使用自然语言意图在表格数据集上运行 plexe

    python -m plexe.main \ --train-dataset-uri data.parquet \ --intent "predict whether a passenger was transported" \ --max-iterations 5

使用前须知

  • 需要付费 LLM 提供商(如 OpenAI 或 Anthropic)的 API 密钥。
  • 官方仅测试了 openai 和 anthropic 模型,其他提供商可能无法使用。
  • 仅支持表格数据(Parquet、CSV、ORC、Avro),需要 Python 3.10 至 3.12。

标签

常见问题

plexe 是什么?
通过自然语言描述需求和表格数据,由多智能体流程自动构建机器学习模型。 plexe 允许用自然语言描述目标来创建机器学习模型。用户提供表格数据(支持 Parquet、CSV、ORC、Avro)和一句任务描述,例如“预测乘客是否被传送”,系统中的 AI 智能体会自动搜索并训练出可用模型。
plexe 可以免费使用吗?用的什么许可证?
plexe 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
plexe 还在维护吗?
超过半年未提交。plexe 最近一次提交是在 2026年3月6日。
plexe 适合用来做什么?
在表格数据上快速原型化预测模型,无需手写训练代码。 生成基线模型,用于与人工构建的流程对比。 将训练好的模型打包,部署到独立的推理环境中。
plexe 怎么安装和运行?
从 PyPI 安装软件包: pip install plexe 设置 LLM 提供商的 API 密钥: export OPENAI_API_KEY=<your-key> export ANTHROPIC_API_KEY=<your-key> 使用自然语言意图在表格数据集上运行 plexe: python -m plexe.main \ --train-dataset-uri data.parquet \ --intent "predict whether a passenger was transported" \ --max-iterations 5
在哪里获取 plexe?
源代码和安装说明在 GitHub 上:https://github.com/plexe-ai/plexe

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目