Skip to content

submarinecantfly/sgd

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

开发环境:win10, python3.6

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages