node package manager
Easy collaboration. Discover, share, and reuse code in your team. Create a free org »



( Development is current suspended ).

A Collection of Machine Learning algorithms built for use with NodeJS

build status


npm install node-ml

Basic API Knowledge

Models are Instantiated with a TrainingSet as an argument

Models are trained via .train()

Models are acted upon by the perceive or predict functions. These functions currently do the same thing, the wording is indicative of the nature of the result, and the action the model has taken on the data.

Models are designed to be trained at the start of a node instance, not during runtime.

Models all inherit from EventEmitter

Models support either Callbacks or Events

callbacks are of the structure function( err, result )

Models support 3 events : trained, response, error

Callbacks override Events, if you specify a callback and event will not be received.

The Single Layer Perceptron

With the single layer perceptron is it possible to solve Linearly Seperable Problems. This makes the SLP a fast tool for solving simple classification problems.

The SLP takes as input a list of 1x2 vectors as in


We must also provide the SLP a list of expected outputs for each vector. Currently the system only supports 1 | -1 These outputs define the side of the line the elements fall on. It is not important which value you give to which inputs. Just that these inputs correspond in a linear way to the outputs.

So for the above input we prove


Now the SLP will solve for where [1,1] === 1 and [-1,-1] === -1

Training the SLP

SingleLayerPerceptron(inputs, outputs, learningRate)

slp = new SingleLayerPerceptron(inputs, outputs, 0.001);
slp.train(function(trainedModel) { 
    trainedModel.perceive([1,1], function(result) {
      //should print out 1 
    trainedModel.perceive([-1,-1], function(result) {
      //should print out -1 

Also Events are supported

current events are: trained, response, error

trained is fired when a model completes training

response is fired when a model completed a perception or prediction phase

error is fired ... well on an error.

    slp = new SingleLayerPerceptron(inputs, outputs, 0.001);
    slp.on('error', function(err) {
    slp.on('trained', function(trainedModel) {
        t.same(true,(trainedModel != undefined));
    slp.on('response', function(response) {
        perceivedTestCount -= 1;
        var result = response.out;
        var input =;
        var expectedIndex;
        for (var i in perceivedTestInput ) {
            if (perceivedTestInput[i].toString() == input.toString())
                expectedIndex = i;
        var expected = perceivedTestOutput[expectedIndex];
        t.same(expected, result);
        if (perceivedTestCount == 0 )

Even Better remember the above trained model is a Line seperating a 2d dimension space from -1 to 1 We can input any value in this range and get an output however this limited training set is a bad choice but heres some outputs

Obtained From Running examples/singlelayerperceptron2.js

Input: 1,1
Input: -1,1
Input: 1,-1
Input: -.5,1
Input: .5,-1
Input: .2,.45634
Input: .2,-.45634
Input: -.4,-.4
Input: -1,-1

The Multi Layer Perceptron

With the Multi Layer Perceptron it is possible to Classify linearly non seperable data set. Meaning that the data fits to a polynomial function.

Refer to examples.

The Linear Regression Model

With Linear Regression we can predict outcomes based on an input.

Refer to examples.

The KMeans Classifier

This implementation of the KMeans classifier is an N-Dimensional classification algorithm. It works by:

  1. Generating Random K Centroids.

  2. Assigning a K centroid to a point p in Training set T such that the Cost(p) with respect to k is minimized; Cost(p) is defined as the distances from P to each K

  3. Move the centroids to the Mean of each cluster assigned to them

  4. Repeat until movement no longer occurs.

Refer to examples for Usage:

Bitdeli Badge