LSTM Prediction Example

by pulipuli

HTML

<script src="https://cdn.rawgit.com/wagenaartje/neataptic/cd0d9e68/dist/neataptic.js"></script>
<script src="https://code.jquery.com/jquery-3.2.1.min.js"></script>

JavaScript

var network = new neataptic.Architect.LSTM(1,6,1);

var trainingData = [
  { input: [0], output: [0] },
  { input: [0], output: [0] },
  { input: [0], output: [1] },
  { input: [1], output: [0] },
  { input: [0], output: [0] },
  { input: [0], output: [0] },
  { input: [0], output: [1] },
];

network.train(trainingData, {
  log: 500,
  iterations: 6000,
  error: 0.03,
  clear: true,
  rate: 0.05,
});

for(var i in trainingData){
  var input = trainingData[i].input;
  var output = Math.round(network.activate([input]));
  $('html').append('<p>Input: ' + input[0] + ', output: ' + output + '</p>');
}

for(var i = 0; i < 10; i++){
  var input = output;
  var output = Math.round(network.activate([input]))
  $('html').append('<p>Next prediction:' + output + '</p>');
}