- Load dataset
- Read image data to Swift array
- Label to one hot
- CPU
- Implement dot multiplication
- Implement transpose
- Create init weight based on gaussian distribution
- Computational graph
- Liner
- Mul layer
- Add layer
- Relu
- Softmax
- SGD
- Loss: Cross-entropy
- Network
- Liner
- UI
- Train
buhe / app-nn Goto Github PK
View Code? Open in Web Editor NEW๐ฆ Visual neural network from scratch
License: GNU General Public License v3.0