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