关于 neural-mcp
neural-mcp 是用于构建、训练和分析神经网络的 MCP 服务,面向机器学习实验。它涵盖数据集加载、训练、评估、调参和模型导出,并可选用 CUDA 加速。
主要功能
- 01构建 ResNet18、MobileNet 和自定义模型
- 02加载 CIFAR-10、MNIST 和 ImageNet
- 03按可配置参数训练模型
- 04评估模型并计算指标
- 05自动搜索超参数
- 06导出为 ONNX 或 TorchScript
适用场景
- 01在标准数据集上训练模型
- 02试验自定义网络架构
- 03衡量模型表现并检查预测
- 04将训练好的模型导出到其他工作流
MCP 相关
向 MCP 客户端提供神经网络模型、数据加载、训练、评估、分析、调参和导出工具。
配置
本地服务的配置在各客户端中通用,复制后粘贴到客户端的 MCP 配置文件即可。
工具
还没有获取到工具列表
本地服务需要在你的电脑上运行后才能读取工具,维护者暂未提供工具列表。
详细介绍
scicomp-neural-mcp
mcp-name: io.github.andylbrummer/neural-mcp
用于神经网络训练和实验的 MCP 服务。
概览
此服务提供用于构建、训练和分析神经网络的工具:
- 模型构建 - 预训练模型(ResNet、MobileNet)和自定义架构
- 数据集加载 - CIFAR-10、MNIST、ImageNet 数据集及标准预处理
- 训练 - 支持配置学习率和批量大小的完整训练循环
- 评估 - 模型评估、指标计算和分析
- 超参数调优 - 自动超参数搜索
- 导出 - 将模型导出为 ONNX 和 TorchScript 格式
- GPU 加速 - 可选的 CUDA 训练加速
安装与使用
# Run directly with uvx (no installation required)
uvx scicomp-neural-mcp
# Or install with pip
pip install scicomp-neural-mcp
# With GPU support (recommended for training)
pip install scicomp-neural-mcp[gpu]
# Run as command
scicomp-neural-mcp
可用工具
模型管理
define_model- 创建神经网络模型(ResNet18、MobileNet、自定义模型)load_pretrained- 从 torchvision 或 Hugging Face 加载预训练模型get_model_summary- 获取逐层详细摘要export_model- 导出为 ONNX 或 TorchScript
数据加载
load_dataset- 加载标准数据集(CIFAR-10、MNIST、ImageNet)create_dataloader- 创建带有随机打乱功能的批处理数据加载器
训练
train_model- 使用可配置参数在数据集上训练模型get_experiment_status- 监控训练进度evaluate_model- 在测试集上评估
分析
compute_metrics- 计算详细的性能指标confusion_matrix- 生成混淆矩阵plot_training_curves- 可视化损失和准确率曲线visualize_predictions- 检查模型对样本的预测结果
超参数优化
tune_hyperparameters- 自动超参数搜索
配置
使用环境变量启用 GPU 加速:
MCP_USE_GPU=1 scicomp-neural-mcp
示例
📖 代码示例
请参阅 EXAMPLES.md 中的实用教程:
- MNIST 数字识别(完整工作流)
- 使用 ResNet 进行迁移学习
- 超参数优化
- 混淆矩阵分析
- 渐进式学习路径(初级 → 高级)
📚 完整文档
完整的 API 参考请参阅 API 文档。
Math-Physics-ML MCP 系统的一部分
这是科学计算综合系统的一部分。完整生态请参阅文档。
评论
支持 Markdown
一个开放协议,规定了 AI 应用如何以统一方式连接外部工具与数据源。客户端负责与模型对话,MCP 服务负责提供工具、资源和提示模板,两者通过标准消息通信,同一个服务可以被不同客户端复用。