JavaScript 中的神经网络

Question

我的神经网络有点问题。我已将其设置为生成一个包含 5 个值的数组； 0 或 1，即 [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)

出于某种原因，当它转到 "guess" 时，它没有给我一个值。有什么想法吗？

Answer 1

它出错的原因有两个

net.run 的输出是一个数组 - 您可能需要其中的第一项。
net.train 中 output 的输入是一个数组 - 您正在向它传递一个不同的值

通过一些更改，您的代码将按照（我认为）您期望的方式运行：

在您的 ask 方法中始终使用 guess[0]

将 oposite 变量用方括号括起来使其成为一个数组

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

下面的工作代码供参考（虽然不能在 stacksnippet 中工作）

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);
    //console.log(array);
    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[0] + ". Am I correct? (Y/N)", (answer) => {

        if(answer == "Y" || answer == "y"){
            ca++;
            net.train([
                {input : array, output : Math.round(guess[0])}
            ]);
        }else if(answer == "N" || answer == "n"){
            wa++;
            var roundGuess = Math.round(guess[0]);
            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();

JavaScript 中的神经网络

Neural Network in JavaScript

javascript

neural-network

brain.js