javascript - Brain.js 根据高度/体重猜测成人还是 child

Question

我正在尝试建立一个简单的 Brain.js 示例来了解它是如何工作的。我用过this tutorial (成人/ child 示例为 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 英尺或以下，则您是 child ；如果您的高度为 6 英尺或以上，则您是成年人。如果您高度 5 英尺，则取决于您的体重。

但是如果我在那里放一个明确的 child (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
  };
}