Мозг. js предсказывает непонятный исход - PullRequest
0 голосов
/ 19 марта 2020

Я пытаюсь поиграть с Brain. js прогнозирование Я нашел пример на этом топике StackOverflow c

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

//Same test as previous, but combined on a single set
const trainingData = [
    [8,8,1],[8,8,3],[8,8,5],[8,2,8],[3,6,6],[8,4,5]
];

net.train(trainingData, { log: true, iterations:200 });

console.log( net.run([[8,2,3]]));

console.log( net.forecast([[8,8,2]], 7)) ;

Над кодом работает и прогнозирует следующее Значения Но когда я делаю небольшие изменения, такие как тренировочные данные и размер ввода, я получаю прогноз, который не имеет никакого смысла для меня

Вот мой код

const brain = require('brain.js');
const net = new brain.recurrent.LSTMTimeStep({
   inputSize: 2,
   hiddenLayers: [4],
   outputSize: 2
});

//Same test as previous, but combined on a single set
const trainingData = [
    [7244.56, 7248.89],[7230.48, 7248.51],[7236, 7234.89],[7239.75, 7244.98],[7234.43, 7230.56], 
    [7243.12,7234.43]
];

net.train(trainingData, { log: true, iterations:20000 });

// console.log( net.run([[7245,7246]]));

console.log( net.forecast([[7238.58, 7240.54]], 1)) ;

и результат

 Float32Array [ -0.05796178802847862, 0.04729847237467766 ]

Я ожидал получить или число от 0 до 1 (%).

Я пытался тренироваться с целыми числами, но это ничего не меняет

ответ Я нашел полезный курс, который объясняет все это курс

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...