Прогнозировать выход для данного входа с мозгом. js? - PullRequest
3 голосов
/ 28 марта 2020

Я новичок в Brain. js и я не очень понимаю, как все работает. У меня есть набор данных с кучей входов и выходов. Найдя шаблон, я хочу, чтобы NN предсказал, какой будет выход для данного входа.

Например, для

  [
     { input: [ 1 ], output: [ 2 ] },
     { input: [ 2 ], output: [ 3 ] },
     { input: [ 3 ], output: [ 4 ] },
     { input: [ 4 ], output: [ 5 ] }
   ]

Я бы хотел, чтобы AI для входа 17 отображал 18. Это был просто пример, это мой фактический набор данных:

[
  { input: [ 0 ], output: [ 198 ] },
  { input: [ 1 ], output: [ 198 ] },
  { input: [ 2 ], output: [ 241 ] },
  { input: [ 3 ], output: [ 252 ] },
  { input: [ 4 ], output: [ 277 ] },
  { input: [ 5 ], output: [ 315 ] },
  { input: [ 6 ], output: [ 366 ] },
  { input: [ 7 ], output: [ 496 ] },
]

И это мой код:

exports.train = data => {

 const network = new brain.NeuralNetwork({ hiddenLayers: [] });

  network.train(data);

  console.log(network.run([2000]));

 };
...