Skip to content

ZhangHanDong/rust-nn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

用 Rust 实现一个神经网络

该项目仅用于学习神经网络

我们将实现一个多层感知机(MLP),并应用于一个二分类问题,模拟识别猫和非猫的图像。

一些优化手段

  1. 训练时 rayon 并行
  2. 隐藏层使用 Relu,输出层使用 Sigmoid
  3. 使用 Adam 算法 (使用策略模式,算法可切换)
  4. 在计算成本时引入 L2 正则化

运行

cargo run --bin train

About

Rust Deep Neural Network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages