JavaScript中的神经网络

时间:2018-03-02 16:12:52

标签: javascript neural-network brain.js

我的神经网络有点麻烦。我已将其设置为生成一个包含5个值的数组; 01,即[1,1,0,1,0]。并且使用Node.js我控制台记录随机数组,如果我回复y,它将使用正确的输出将其添加到训练中,反之亦然。一旦我做出响应,genRan()就会运行并创建一个新的随机数组并保存" guess"到var guess。但是,在第一次运行后,它不再给我一个猜测值,而是:[object Object]

以下是代码:

var brain = require('brain.js');
var net = new brain.NeuralNetwork();
const readline = require('readline');

const r1 = readline.createInterface({
  input: process.stdin,
  output: process.stdout
});

var ca = 0,
    wa = 0;

net.train([
    {input: [0,0,0,0,0], output: [0]}
]);

function genRan(){
    var a,b,c,d,e;
    var array = [];
    a = Math.round(Math.random());
    b = Math.round(Math.random());
    c = Math.round(Math.random());
    d = Math.round(Math.random());
    e = Math.round(Math.random());

    array.push(a,b,c,d,e);
    var guess = net.run(array);
    ask(array,guess);
}

function ask(a,b){
    var array = a,
        guess = b;
    r1.question((wa+ca) + ") input: " + array + " We think: " + guess + ". Am I correct? (Y/N)", (answer) => {

        if(answer == "Y" || answer == "y"){
            ca++;
            net.train([
                {input : array, output : Math.round(guess)}
            ]);
        }else if(answer == "N" || answer == "n"){
            wa++;
            var roundGuess = Math.round(guess);
            var opposite;
            switch (roundGuess){
                case 1:
                    opposite = 0;
                    break;
                case 0:
                    opposite = 1;
                    break;
                default:
                    opposite = null
            }
            net.train([
                {input : array, output : opposite}
            ]);     
        }
        console.log("Success percent: " + (100 *ca/(ca+wa)) + "% " + (ca+wa) +" attempts\n\r");
        genRan();
    })

}
genRan();

第一个问题很好,并提出了这个问题:

0) input: 0,0,0,0,0 We think: 0.07046. Am I correct? (Y/N)

当我回复时,我得到:

Success percent: 100% 1 attempts

1) input 1,1,1,0,1 We think: [object Object]. Am I correct? (Y/N)

出于某种原因,当它进入"猜测"它没有给我一个价值。有什么想法吗?

1 个答案:

答案 0 :(得分:2)

出错的原因是双重的

  1. net.run的输出是一个数组 - 你可能想要它的第一个项目。
  2. outputnet.train的输入是一个数组 - 您为其传递了一个不同的值
  3. 通过一些更改,您的代码可以正常工作(我认为):

    1. 在{/ 1}}方法中使用guess[0]
    2. ask变量换成方括号以使其成为数组

      oposite
    3. 下面的工作代码供参考(虽然不能在stacksnippet中使用)

      
      
       net.train([
              {input : array, output : [opposite]}
          ]);