Как обучить Brain.js, чтобы получить желаемый результат из массива чисел - PullRequest
Новая
       12

Как обучить Brain.js, чтобы получить желаемый результат из массива чисел

0 голосов
/ 10 мая 2019

Я смотрел на YouTube несколько обучающих видео о том, как начать машинное обучение с использованием Brain.js.

Я рад попробовать какой-нибудь код ....

Итак, для начала я попытался передать список чисел в качестве входных данных для NeuralNetwork и обучить его, чтобы я мог получить желаемый результат ...

Вот мой код .. 

const brain = require('brain.js');
const net = new brain.recurrent.LSTMTimeStep();
    net.train([
      {input:[1,2],output:[2]},
      {input:[2,1],output:[2]},  
      {input:[2,3],output:[1]},
      {input:[3,2],output:[1]},
      {input:[1,3],output:[3]}   
    ],{
        log: (error) => console.log(error)
    });

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

Выход:

итераций: 19920, ошибка обучения: 0,2050450325012207 итерации: 19930, ошибка обучения: 0,20473779042561846 итерации: 19940, ошибка обучения: 0,20434550444285077 итераций: 19950, ошибка обучения: 0,201515732065836587 итерации: 19960, ошибка обучения: 0,201212214597066242 итерации: 19970, ошибка обучения: 0,20421248674392706 итерации: 19980, ошибка обучения: 0,20441961288452148 итерации: 19990, ошибка обучения: 0,20479104121526084

+2,0533313751220703

Я ожидаю, что выход будет 3 ... но он показывает 2,05 ...

Что нужно сделать, чтобы получить желаемый результат

...