A small TypeScript-only Neural Networks library with zero dependencies. It requires Node.js to run, but it can also run in the browser with a few modifications.
You need Node.js and npm installed on your computer
- FeedForward neural network with momentum and learning rate
- Dense, Activation and Flatten Layers
- mse, bce, ce loss functions
- linear, step, tanh, sigmoid, softmax, ReLu, leakyReLu activation functions
- ad-hoc linear algebra library
-
Install dependencies
npm i
-
Download example datasets from here and place extracted data in
datasets -
Run examples
XOR
npm run xorMNIST
npm run mnist
- Convolutional Layers
- Better logging