Skip to content

Simple Neural Network基于 NumPy 从零实现神经网络

从数学原理到代码实战 — 覆盖线性层、激活函数、损失函数、反向传播、梯度下降的完整训练管线

项目总览

Simple Neural Network 是一个纯 NumPy 实现的神经网络练习项目,旨在以最清晰的方式展示神经网络的核心计算流程。

已实现模块

模块实现
网络层LinearLayer(全连接,Xavier 初始化)、ReLULayerSigmoidLayerTanhLayer
损失函数MSELoss(回归)、CrossEntropyLoss(分类,内置 Softmax)
模型SequentialModel(顺序容器,继承 BaseLayer)
优化器SGDOptimizer(随机梯度下降)
训练器Trainer(mini-batch、shuffle、eval、predict、epoch 日志)
评估指标分类准确率、回归均方误差
数据DataGenerator(XOR/Spiral/Sine)、DataLoader
持久化CheckpointIO(模型参数 .npz 读写)

快速命令

bash
# 生成数据集
python main.py

# 运行全部测试
pytest tests -q

# 启动文档开发服务器
npm run docs:dev

# 构建文档
npm run docs:build