跳到正文
Hacker News· AryanDinakaran1·· 6 天前AI 评分58

EpochGo 发布本地优先多智能体 AutoML 平台,基于 Ollama 零 API 成本

Show HN: EpochGo – Local-first multi-agent AutoML on Ollama ($0 API costs)

AI 导读

Epochlypse Research 开源本地优先的多智能体 AutoML 平台 EpochGo,基于 CrewAI、Ollama 与 Scikit-Learn/XGBoost/LightGBM/CatBoost,无需 API key 和云成本,全部在本地离线运行。

正文 · AI 翻译

自主、本地优先的多智能体 AutoML 平台
由 Epochlypse Research 开发 • 基于 CrewAI、Ollama 和 Scikit-Learn / XGBoost / LightGBM / CatBoost

Docker CrewAI Ollama FastAPI License: MIT

零 API 密钥。零云成本。100% 私密且离线运行在您的本地机器上。

EpochGo Live Demo


🌟 概述

EpochGo 是由 Epochlypse Research 开发的开源、本地优先的 AutoML 软件系统。只需 git clone 仓库并运行 Docker。您将在浏览器中获得一个本地化链接(http://localhost:8000),在那里您可以创建项目、放入原始 CSV 数据集,并让一个由 三个自主 AI 智能体 组成的协调团队完成所有数据科学工作:

  1. 🧹 智能体 1:数据预处理与特征架构师

    • 自动检测模式、缺失值、极端异常值(IQR)、分类编码、缩放和高维多重共线性(PCA)。
    • 生成并执行独立的 Scikit-Learn 管道(preprocessing.py)。
    • 生成详细的处理前后对比报告(preprocessing.md)。
    • 保存序列化转换器(preprocessor.joblib)和清洗后的数据集(cleaned_data.csv)。
  2. 🏆 智能体 2:机器学习基准测试与调优专家

    • Evaluates 7 baseline machine learning algorithms without hyperparameter bias:
      • 分类:XGBoost、LightGBM、CatBoost、随机森林、Extra Trees、梯度提升、逻辑回归。
      • 回归:XGBoost、LightGBM、CatBoost、随机森林、Extra Trees、梯度提升、岭回归。
    • 评估基线准确率/F1/R²/RMSE,并选出 前三名竞争者。
    • 运行高效超参数优化(RandomizedSearchCV)。
    • 生成可复现的训练脚本(model_building.py)和全面的 Markdown 报告(model_building.md)。
    • 将冠军模型序列化到 model.joblib。
  3. 🚀 智能体 3:MLOps 与模型服务工程师

    • 合成一个生产就绪的独立 FastAPI 微服务脚本(serve.py),包含与您的数据集模式匹配的 Pydantic 类型验证。
    • 启用交互式浏览器内 预测游乐场,实时测试预测和概率。
    • 生成即用型 curl 和 Python API 代码片段。

🚀 快速开始(单条命令)

先决条件

  • 已安装 Docker 和 Docker Compose。
  • 建议 16 GB 内存(Apple Silicon M1/M2/M3/M4/M5 Mac,或具有 16GB 内存的 Windows/Linux)。

1. 克隆并运行

git clone https://github.com/AryanDinakaran/EpochGo.git
cd EpochGo
docker compose up --build

2. 打开浏览器

导航至:

http://localhost:8000

就是这样!EpochGo 将自动连接到 Ollama,确保编码模型已加载,并启动 Web 仪表板。


💻 推荐的本地模型与硬件基准

指标 目标规格
推荐模型 qwen2.5-coder:7b
为什么选择这个模型? 世界一流的代码合成、scikit-learn 精通以及精确的 Markdown 报告撰写。
显存 / 内存占用 约 4.7 GB 内存
目标硬件 16 GB 内存 Mac(Apple Silicon M 系列)或 PC
可用余量 剩余 >10 GB 内存用于并行 XGBoost / CatBoost / LightGBM 模型训练
低资源回退 qwen2.5-coder:1.5b 或 qwen2.5-coder:3b 适用于 8GB 环境

提示

您的 Mac 上已经安装了 Ollama? 您可以通过将 EpochGo 指向您现有的 Ollama 实例来节省磁盘空间并运行原生 Metal GPU 加速:

OLLAMA_BASE_URL=http://host.docker.internal:11434 docker compose up

🏗️ 架构与智能体流水线

flowchart TD
    A["Raw CSV Upload & Project Name"] --> B["Agent 1: Data Preprocessor"]

    subgraph Agent1 ["🧹 Agent 1: Data Preprocessing"]
        B --> B1["Data Profiling & Pattern Analysis"]
        B1 --> B2["Imputation, Outlier Clipping, Encoding & Scaling"]
        B2 --> B3["Synthesize & Execute preprocessing.py"]
        B3 --> B4["Generate preprocessing.md & preprocessor.joblib"]
    end

    B4 --> C["Agent 2: Model Builder"]

    subgraph Agent2 ["🏆 Agent 2: Model Building & Tuning"]
        C --> C1["7 Baseline Algorithms Benchmark"]
        C1 --> C2["Select Top 3 Best Performers"]
        C2 --> C3["RandomizedSearchCV Hyperparameter Tuning"]
        C3 --> C4["Save Champion model.joblib"]
        C4 --> C5["Generate model_building.py & model_building.md"]
    end

    C5 --> D["Agent 3: Model Serving"]

    subgraph Agent3 ["🚀 Agent 3: Deployment & Serving"]
        D --> D1["Extract Pydantic Input Schema"]
        D1 --> D2["Synthesize Standalone serve.py (FastAPI)"]
        D2 --> D3["Activate In-Browser Inference Playground"]
    end

加载中


📂 项目输出与生成的工件

您创建的每个项目都会在 projects/<project_id>/ 内存储完整、可复现的工件:

文件 用途
raw_data.csv 原始上传的数据集
cleaned_data.csv 经过缺失值、异常值、缩放和编码处理后的转换数据集
preprocessor.joblib 用于下游推理的 Scikit-Learn transformer 流水线
preprocessing.py 独立、可复现的数据准备 Python 脚本
preprocessing.md 全面的 Markdown 报告,剖析前后分布
model.joblib 训练好的冠军机器学习模型产物
model_building.py 独立、可复现的模型训练与评估脚本
model_building.md 基准排行榜(7 个模型)、调优统计和特征重要性
serve.py 可部署到任何地方的生产级 FastAPI 微服务
metadata.json 项目状态、指标记分板和特征模式

🧪 包含的示例数据集

EpochGo 在 sample_data/ 中预置了两个数据集,可立即用于测试:

  1. 客户流失(sample_data/customer_churn.csv):

    • 任务:二分类(churn:0 或 1)
    • 特征:age、tenure_months、monthly_charges、total_charges、contract、internet_service 等。
    • 包含真实世界的缺失值和异常值,以演示智能体清洗。
  2. 房价(sample_data/house_prices.csv):

    • 任务:回归(price)
    • 特征:square_feet、bedrooms、bathrooms、year_built、neighborhood、condition 等。

🛠️ 本地开发(不使用 Docker)

如果你更愿意直接在宿主机上运行 EpochGo:

# 1. Clone repository
git clone https://github.com/AryanDinakaran/EpochGo.git
cd EpochGo

# 2. Create virtual environment
python3 -m venv venv
source venv/bin/activate

# 3. Install dependencies
pip install -r requirements.txt

# 4. Make sure Ollama is running
ollama serve
ollama pull qwen2.5-coder:7b

# 5. Start EpochGo
uvicorn epoch_go.main:app --host 0.0.0.0 --port 8000 --reload

📡 API 参考

  • GET /api/health - 检查服务状态和 Ollama 模型连接性。
  • GET /api/projects - 列出所有项目及执行状态。
  • POST /api/projects - 上传 CSV、命名项目并触发 CrewAI 流水线。
  • GET /api/projects/{id} - 获取项目元数据、状态和基准排行榜。
  • GET /api/projects/{id}/files/{filename} - 获取生成的 Markdown 报告和 Python 脚本。
  • GET /api/projects/{id}/download/{filename} - 下载 .joblib 模型或 .py 脚本。
  • POST /api/projects/{id}/predict - 针对项目的冠军模型运行实时预测。
  • WS /ws/projects/{id} - 用于智能体推理日志的实时 WebSocket 流。

🏛️ 维护与开发方

Epochlypse Research


📄 许可证

MIT 许可证。可免费用于个人、商业和研究用途。

来源:Hacker News · github.com