Dann.js

Deep Neural Network Library for JavaScript.

matiasvlevi/Dann dannjs.org dannjs

README

Deep Neural Network Library for Javascript

Train a neural network with your data & save its trained state!

Demo • Installation • Documentation • Contribute • Discord • License

Installation

CDN :

``` html
<script src="https://cdn.jsdelivr.net/gh/matiasvlevi/dann@v2.4.1e/build/dann.min.js"></script>
```

Node :

```
npm i dannjs
```

dannjs on npmjs.com

Getting started

Require package

Components from the library can be imported like this

``` js
const { Dann } = require('dannjs');
```

Basic model construction

Setting up a small (4,6,6,2) neural network.

``` js
const nn = new Dann(4, 2);
nn.addHiddenLayer(6, 'leakyReLU');
nn.addHiddenLayer(6, 'leakyReLU');
nn.outputActivation('tanH');
nn.makeWeights();
nn.lr = 0.0001;
nn.log({details:true});
```

Train by backpropagation

Training with a dataset.

``` js
//XOR 2 inputs, 1 output
const dataset = [
{
input: [0, 0],
output: [0]
},
{
input: [1, 0],
output: [1]
},
{
input: [0, 1],
output: [1]
},
{
input: [1, 1],
output: [0]
}
];
//train 1 epoch
for (data of dataset) {
nn.backpropagate(data.input, data.output);
console.log(nn.loss);
}
```

Train by mutation

For neuroevolution simulations. Works best with small models & large population size.

``` js
const populationSize = 1000;
let newGeneration = [];
for (let i = 0; i < populationSize; i++) {
// parentNN would be the best nn from past generation.
const childNN = parentNN;
childNN.mutateRandom(0.01, 0.65);
newGeneration.push(childNN);
}
```

Standalone function

Convert a Neural Network to a JS function that can output predictions without the library.

``` js
let strfunc = nn.toFunction();
console.log(strfunc);
```

Save JSON

``` js
let json = nn.toJSON();
console.log(json);
```

Demo:

[AI predicts San-francisco Housing prices.](https://dannjs.org/livedemo.html)

more examples & demos here

Online editor:

https://dannjs.org/sandbox

Socials

Graph Dann models with this library

Dann-p5

Stickers

Get Dannjs stickers!

Contributors

_{Matias Vazquez-Levi} 💻 📖 ⚠️ ✅	_{Francesco Ciulla} 📢	_Labnan 🐛 💻 ⚠️	_sharkAce 💻	_{Hasnain Iqbal} 💻 ⚠️	_{EL Ramos} 🐛 ⚠️ 💻	_viabhinav ✅
_and1can 💻 ⚠️

Any contributions are welcome! See CONTRIBUTING.md.

License

MIT