n42
n42 is the deep learning module for nodejs.
How to install
$ npm install n42
Getting started
var n42 = require('n42');
var input = $M([
[1.0, 1.0, 0.0, 0.0],
[1.0, 1.0, 0.2, 0.0],
[1.0, 0.9, 0.1, 0.0],
[0.0, 0.0, 0.0, 1.0],
[0.0, 0.0, 0.8, 1.0],
[0.0, 0.0, 1.0, 1.0]
]);
var label = $M([
[1.0, 0.0],
[1.0, 0.0],
[1.0, 0.0],
[0.0, 1.0],
[0.0, 1.0],
[0.0, 1.0]
]);
var sda = new n42.SdA(input, label, 4, [3, 3], 2);
sda.pretrain(0.3, 0.01, 1000);
sda.finetune(0.3, 50);
var data = $M([
[1.0, 1.0, 0.0, 0.0],
[0.0, 0.0, 1.0, 1.0]
]);
console.log(sda.predict(data));
Algorithms
Class | Implemented algorithm |
---|
NN | Neural Network |
LogisticsRegression | Logistics Regression |
SdA | Stacked denoised Autoencoder |
DBN | Deep Belief Nets |
API Docs
n42 API Doc
LICENSE
MIT License. Please see the LICENSE file for details.