Question

我正在尝试建立一个简单的brain.js示例，以了解它是如何工作的。我使用this tutorial（成人/儿童示例为5:10）。我的代码如下所示：

var brain = require("brain.js");

const network = new brain.NeuralNetwork();

network.train([
  {input:{height:1, weight:20},output:{adult:0}},
  {input:{height:2, weight:40},output:{adult:0}},
  {input:{height:3, weight:60},output:{adult:0}},
  {input:{height:4, weight:80},output:{adult:0}},
  {input:{height:5, weight:100},output:{adult:0}},
  {input:{height:5, weight:200},output:{adult:1}},
  {input:{height:6, weight:220},output:{adult:1}},
  {input:{height:7, weight:240},output:{adult:1}},
  {input:{height:8, weight:260},output:{adult:1}},

]);


const testPerson = network.run({height:1,weight:20})
console.log(testPerson)

很明显，如果你是4英尺或更小，你就是一个孩子，如果你是6英尺或更长，你就是一个成年人。如果你是5英尺，那么这取决于你的体重。

但是如果我把一个聪明的孩子放在那里（testPerson），我会得到回复：

{ adult: 0.4641557037830353 }

似乎基于这些数据，它应该更加自信1英尺20磅的人不是成年人。

我在这里想念的软件是如何工作的？

Answer 1

您只需要对输入进行规范化：https://jsfiddle.net/robertleeplummerjr/jn056e7L/5/

const network = new brain.NeuralNetwork();

network.train([
  {input:{height:1, weight:20},output:{child:1}},
  {input:{height:2, weight:40},output:{child:1}},
  {input:{height:3, weight:60},output:{child:1}},
  {input:{height:4, weight:80},output:{child:1}},
  {input:{height:5, weight:100},output:{child:1}},
  {input:{height:5, weight:200},output:{adult:1}},
  {input:{height:6, weight:220},output:{adult:1}},
  {input:{height:7, weight:240},output:{adult:1}},
  {input:{height:8, weight:260},output:{adult:1}},
].map(normalize))


const testChild = network.run(normalizeInput({height:2,weight:20}))
console.log('a child', testChild)
const testAdult = network.run(normalizeInput({height:5,weight:200}))
console.log('an adult', testAdult)

function normalize(item) {
  return {
    input: normalizeInput(item.input),
    output: item.output
  }
}

function normalizeInput(input) {
    return {
    height: input.height / 10,
    weight: input.weight / 300
  };
}

Brain.js根据身高/体重猜测成人或儿童

1 个答案: