Используя мозг. js, чтобы предсказать будущее значение - PullRequest
Новая
       36

Используя мозг. js, чтобы предсказать будущее значение

0 голосов
/ 18 марта 2020

Я относительно новичок в использовании Javascript, и я пытаюсь использовать мозг. js, чтобы взять набор значений из файла CSV и предсказать, какое следующее значение будет основано на значениях. Все еще учусь и знаю, что мой код не так хорош, как мог бы, но код работает не так, как ожидалось (прогнозируемое значение не печатается, как видно со ссылкой на document.write ()), и не уверен, что делать, чтобы разрешить его работать таким образом. Любой совет или помощь приветствуются. Спасибо! 

<html>
<body>
<script src = "https://cdnjs.cloudflare.com/ajax/libs/d3/5.7.0/d3.min.js"></script>
<script>
function makeChart(cases) {
  var caseDates = cases.map(function(d) {
    return d.date;
  });
  var worldData = cases.map(function(d) {
    return +d.World;
  });

  var a = worldData;
function scaleDown(step){ // normalize
  return {
    a: step.a / 100000
  };
}


function scaleUp(step){ // denormalize
  return {
    a: step.a * 100000
  };
}


const scaledData = rawData.map( scaleDown);

const trainingData = [
  scaledData.slice(0, 5),
  scaledData.slice(5, 10),
  scaledData.slice(10, 15),
  scaledData.slice(15,20)
];

const net = new brain.recurrent.LSTMTimeStep({
  inputSize: 1,
  hiddenLayers: [8, 8],
  outputSize: 1
});

net.train(trainingData, { 
  learningRate: 0.005, 
  errorThresh: 0.02, 
});

document.write(scaleUp(net.run(trainingData[0])));

// Request data using D3
d3.csv(***csv file***).then(makeChart);

</script>

</body>
</html>
...