AutoForge¶
讓 AI 自己訓練 AI。 AutoForge 會看你的硬體和資料,挑選合適的訓練方法,然後在你自己的電腦上訓練、調參,最後把模型部署成 API 或 MCP 工具。
你可以用四種方式操作它:
| 方式 | 適合 |
|---|---|
網頁介面:autoforge gui |
不想打指令 |
一句話訓練:autoforge train "預測客戶流失" --data data.csv |
讓 AI 代理全程處理(Claude、OpenAI、Ollama 本地模型) |
MCP:autoforge mcp |
已經在用 Claude Desktop / Claude Code 等訂閱制 AI |
CLI:autoforge run start plan.yaml --json |
寫程式、自動化 |
安裝¶
pip install autoforge-trainer # 表格、文字、時間序列、異常偵測(CPU 就能跑)
pip install "autoforge-trainer[deep]" # 加上 Transformer、LLM 微調與 DPO、影像、語音、圖像生成(NVIDIA GPU)
pip install "autoforge-trainer[all]" # 再加上強化學習、Claude 代理和 ONNX 匯出
需要 Python 3.11 以上。GPU 只支援 NVIDIA(CUDA)。
第一次使用¶
autoforge quickstart
這一個指令會依序完成:檢查硬體 → 安裝方法包 → 訓練客戶流失模型 → 註冊模型 → 預測,大約一分鐘。在全新的電腦上(只有 Python,沒有 git),包含 pip install 總共約 63 秒。
Docker¶
docker run --rm -p 127.0.0.1:8800:8800 -v autoforge:/data ghcr.io/honguan/autoforge:cpu # CPU
docker run --rm --gpus all -p 127.0.0.1:8800:8800 -v autoforge:/data ghcr.io/honguan/autoforge:cuda # NVIDIA GPU
接著打開 http://127.0.0.1:8800/。執行其他指令:docker run --rm -v autoforge:/data ghcr.io/honguan/autoforge:cpu quickstart
內建方法包¶
| 方法包 | 用途 | 硬體 |
|---|---|---|
tabular-classic |
表格分類、回歸 | CPU |
text-classification |
文字分類(TF-IDF、Transformer 微調) | CPU / GPU |
llm-finetune |
LLM 微調(LoRA / QLoRA) | GPU 8GB+ |
image-classification |
圖片分類(timm) | GPU 建議 |
timeseries-forecast |
時間序列預測(單一或多條序列) | CPU |
anomaly-detection |
異常偵測 | CPU |
object-detection |
物件偵測(Faster R-CNN) | GPU 建議 |
image-segmentation |
影像分割(LR-ASPP / DeepLabV3) | GPU 建議 |
embedding-finetune |
文字向量微調(搜尋、RAG) | CPU / GPU |
speech-recognition |
語音辨識(Whisper 微調) | GPU 建議 |
llm-preference |
LLM 偏好對齊(DPO) | GPU 8GB+ |
diffusion-lora |
圖像生成 LoRA(Stable Diffusion) | GPU 8GB+ |
reinforcement-learning |
強化學習(stable-baselines3) | CPU / GPU |
recommendation |
推薦系統(ALS、Item-KNN) | CPU |
自己做方法包,或讓 AI 把 GitHub 上的訓練專案轉成方法包,請看方法包格式。
隱私¶
用 autoforge data mark <路徑> -s sensitive 標記的資料,只有本地模型(例如 Ollama)或人可以使用;雲端 AI 對一般資料只看得到統計數字。詳見 MCP 隱私。