node-red-contrib-brain-js

This node provides neural networks. It is based on brain.js

Install

From your node-red directory:

npm install node-red-contrib-brain-js

or

in the Node-red, Manage palette, Install node-red-contrib-brain-js

Usage

This node provides neural networks.
At first you need to see the brain.js documentation

The following neural networks are available :

• 0: brain.NeuralNetwork - Feedforward Neural Network with backpropagation

• 1: brain.recurrent.RNNTimeStep - Time Step Recurrent Neural Network or "RNN"

• 2: brain.recurrent.LSTMTimeStep - Time Step Long Short Term Memory Neural Network or "LSTM"

• 3: brain.recurrent.GRUTimeStep - Time Step Gated Recurrent Unit or "GRU"

• 4: brain.recurrent.RNN - Recurrent Neural Network or "RNN"

• 5: brain.recurrent.LSTM - Long Short Term Memory Neural Network or "LSTM"

• 6: brain.recurrent.GRU - Gated Recurrent Unit or "GRU"

Training

The training data need to be stored in msg.trainData and the format is described here.
The network options can be provided in msg.neuralNetworkOptions.

When the training is done, the network is available in msg.net and can be stored to be imported next time.

Run

The running data need to be stored in msg.runData.

The result is available in msg.decision.

Import network

When the input message contains msg.netJSON, the network is loaded from the JSON provided.

License

MIT License