📐
数学优先
每个模块从数学定义开始,逐步推导到代码实现。包含完整的 LaTeX 公式——前向传播、链式法则求导、梯度下降更新公式。
Simple Neural Network 是一个纯 NumPy 实现的神经网络练习项目,旨在以最清晰的方式展示神经网络的核心计算流程。
| 模块 | 实现 |
|---|---|
| 网络层 | LinearLayer(全连接,Xavier 初始化)、ReLULayer、SigmoidLayer、TanhLayer |
| 损失函数 | MSELoss(回归)、CrossEntropyLoss(分类,内置 Softmax) |
| 模型 | SequentialModel(顺序容器,继承 BaseLayer) |
| 优化器 | SGDOptimizer(随机梯度下降) |
| 训练器 | Trainer(mini-batch、shuffle、eval、predict、epoch 日志) |
| 评估指标 | 分类准确率、回归均方误差 |
| 数据 | DataGenerator(XOR/Spiral/Sine)、DataLoader |
| 持久化 | CheckpointIO(模型参数 .npz 读写) |
# 生成数据集
python main.py
# 运行全部测试
pytest tests -q
# 启动文档开发服务器
npm run docs:dev
# 构建文档
npm run docs:build