dsh-ai-model-builder
Manifest validDSH 左侧栏插件:配置神经网络(学习率/神经元/层数/可选数据集),一键生成 PyTorch 模型架构,并在新会话中自动启动训练
dsh-ai-model-builder
DSH 左侧栏插件:AI 模型构建器。在图形界面里配置一个多层全连接神经网络, 一键生成 PyTorch 模型架构,并一键新开 DSH 会话让 Agent 按数据集自动训练该模型。
面板功能
在 DSH 左侧栏点击「AI 模型构建」图标打开配置面板:
| 区域 | 字段 | 说明 |
|---|---|---|
| 模型结构 | 学习率 | 优化器学习率 |
| 输入维度 / 输出维度 | 数据维度与任务输出 | |
| 每层神经元数量 | 每个隐藏层的神经元数 | |
| 神经网络层数 | 隐藏层数量(1–128) | |
| 数据集与用途 | 数据集路径(可选) | CSV / 图片目录等;留空时训练会生成合成演示数据 |
| 模型主要用途 | 写入 params.json,并作为训练指令的背景 | |
| 保存路径 | 模型文件保存路径 | 「运行」在此写入 model.py |
| 参数文件保存路径 | 「运行」在此写入 params.json |
三个按钮:
- 确认保存 —— 把当前配置持久化到
<DSH_HOME>/data/dsh-ai-model-builder/config.json; - 运行(生成模型架构) —— 保存配置,并生成:
model.py:PyTorch 多层感知机(build_model()读取params.json构建),params.json:学习率 / 层数 / 神经元 / 维度 / 数据集 / 用途等超参数,architecture.json:架构说明与参数量估算(与model.py同目录);
- 训练(新会话自动启动) —— 保存配置后,用 DSH 的 agents 服务新建一个会话
(工作目录 = 模型文件所在目录),并自动注入训练指令:Agent 会检查/安装依赖
(torch、numpy、pandas、scikit-learn)、编写
train.py、按数据集训练、把权重保存到model_weights.pt,最后汇报训练指标。面板随后自动切到该会话,可实时观察执行过程。
HTTP 端点(Node half,仅同源可写)
GET /ai-model-builder/config—— 读取已保存配置POST /ai-model-builder/config—— 保存配置(body:{config: {...}})POST /ai-model-builder/generate—— 保存配置并生成模型架构文件POST /ai-model-builder/train—— 保存配置、新建会话并注入训练指令,返回{sessionId, cwd, ...}
安装(本地 link)
- 把本目录放到
%USERPROFILE%\.dsh\plugins\dsh-ai-model-builder; - 编辑
%USERPROFILE%\.dsh\profiles\<profile>\package.json:dependencies增加"dsh-ai-model-builder": "link:C:/Users/<你>/.dsh/plugins/dsh-ai-model-builder"(用绝对路径,正斜杠);dsh.profile.bundles追加"dsh-ai-model-builder";
- 在 profile 目录执行
pnpm install; - 重启 DeepSeek Harness。
卸载
从 profile package.json 移除依赖与 bundle 条目,pnpm install,重启 DSH;
再删除本插件目录与 <DSH_HOME>/data/dsh-ai-model-builder 即可。
实现说明
- 零外部 npm 依赖:Node half 只 import
node:内置模块;installModelSelection/createUserMessage是按宿主@deepseek-ai/dsh-agent/@deepseek-ai/dsh-llm公开语义内联的最小等价实现 (与官方 headless runner 使用同一调用契约)。 - 客户端通过
main槽位(key=ai-model-builder)+sidebar.panellist槽位 注册左侧栏入口;训练跳转使用sessions.retain(sessionId, {source: "mainView"})layout.selectPanel(null)。
Comments
Loading…
Similar plugins
by Kazusa1085
DSH插件:提供“模型能力”设置页面,用于声明模型输入模态、上下文窗口、输出上限及图像配额,无需手动编辑 `settings.yaml` 文件。
★ 0
MIT
JavaScript
Sep 9, 2026
dsh plugin --profile web add dsh-awesome-model-settingby KouzakiUmi
DSH 插件:从 pi-ai 上游目录发现和管理模型,支持人工确认、按 provider 开关、安全写入与回滚;models.dev 可作可选备用数据源。
★ 0
MIT
JavaScript
Oct 22, 2026
dsh plugin --profile web add dsh-model-refreshby ChisaAlter
DSH(DeepSeek Harness)右侧悬浮面板坞插件:AI 在对话里生成面板,可平铺组合;支持本机/HTTP 数据源、AI 刷新、面板内 dsh.run 模型调用。
★ 3
MIT
TypeScript
Oct 29, 2026
dsh plugin --profile web add dsh-padby jerry9692
DSH 插件:把 workbuddy2api-panel 的模型(倍率/国内国际/思考档位)同步进 DeepSeek Harness,并显示额度与用量
★ 1
MIT
JavaScript
Oct 19, 2026
dsh plugin --profile web add dsh-workbuddy2api-panelby webkubor
给 DSH 模型页补上官方适配器缺的那半:网关可达性探测、模型目录拉取、DeepSeek 余额常驻。零运行时依赖,不改动 DSH 安装。
★ 2
↓ 25/wk
MIT
JavaScript
Oct 18, 2026
dsh plugin --profile web add dsh-llm-hubby Guyao146
DSH 中转站模型思考强度配置与自动识别插件
★ 4
LGPL-2.1
JavaScript
Oct 5, 2026
dsh plugin --profile web add dsh-better-model-thinking-control