node.js - 获取不需要的 Brain.js 输出

Question

我正在尝试学习 Brain.js。我编写了一段代码来输入文本并获取数字作为输出。但我总是得到 NaN 作为输出。

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

net.train([
  {input: "", output:[0]},
  {input: "Jack", output:[1]},
  {input: "Tim", output: [0]},
  {input: "James", output: [0]},
  {input: "JOHN", output: [0]},
  {input: "cathy", output: [0]},
  {input: "Boom", output: [0]},
]);

console.log("Jack = "+net.run("Jack"));
console.log("JOHN = "+net.run("JOHN"));
console.log("cathy = "+net.run("cathy"));

Answer 1

您的输出很好，但您使用的输入标记(字符串)训练 brain.NeuralNetwork 的方法不兼容。你需要以某种方式输入数字。实现此目的的一种方法是使用属性为数字的对象。这将起作用:

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

net.train([
  {input: { "": 1 }, output:[0]},
  {input: { "Jack": 1 }, output:[1]},
  {input: { "Tim": 1 }, output: [0]},
  {input: { "James": 1 }, output: [0]},
  {input: { "JOHN": 1 }, output: [0]},
  {input: { "cathy": 1 }, output: [0]},
  {input: { "Boom": 1 }, output: [0]},
]);

console.log("Jack = "+net.run({ "Jack": 1 }));
console.log("JOHN = "+net.run({ "JOHN": 1 }));
console.log("cathy = "+net.run({ "cathy": 1 }));

工作示例:https://jsfiddle.net/robertleeplummerjr/xz06ghfp/3/