Skip to content

Latest commit

 

History

History
10 lines (7 loc) · 547 Bytes

README.md

File metadata and controls

10 lines (7 loc) · 547 Bytes
  • 主函数: SGD_nn.py
  • 不使用框架,基于矩阵运算手撕随机梯度下降实现非线性分类器
  • 主函数在SGD_nn中,二层网络包含50个神经元默认在3000个非线性可分数据上训练20000轮后绘制训练结果和loss曲线。
  • 超参数以及数据集大小等参数封装在config.py中。
  • network.py中封装了一个二层神经网络类,包含训练、预测等方法
  • tools.py中封装了包括画图、数据集生成、函数计算等工具函数。

开发环境:win10, python3.6