This node provides neural networks. It is based on brain.js
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
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" -
5:
brain.recurrent.LSTM
- Long Short Term Memory Neural Network or "LSTM"
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.
The running data need to be stored in msg.runData.
The result is available in msg.decision.
When the input message contains msg.netJSON, the network is loaded from the JSON provided.
MIT License
This is fork of the fchanson/node-red-contrib-neuralnet, all credits to the original author.