prediction.js

it('can predict a phrase when given the first letter', () => {
  const phrase = 'bob';
  const vocab = new Vocab(['b', 'o']);
  const net = new GRU({
    inputSize: 3,
    inputRange: vocab.characters.length,
    outputSize: 3
  });
  for (var i = 0; i < 100; i++) {
    //train
    net.run(vocab.toIndexes(phrase));
    
    //visual feedback
    if (i % 10 === 0) {
      console.log(vocab.toCharacters(net.predict()).join(''));
    }
  }
  
  //test when given "b", should add "ob" to it
  assert.equal(vocab.toCharacters(net.predict(vocab.toIndexes('b'))).join(''), phrase);
});