javascript - 使用brainjs训练输出不在特定范围内

Question

我是神经网络的新手，我正在尝试训练几个简单的数据点。我有训练输入 x 和 y，我希望输出 c 为：c<x & c>y。我尝试选择 x 和 y 作为特征，但 c 并不总是在我想要的范围内。我还可以使用哪些其他功能？我在 javascript 中使用 Brain.js 库。

Answer 1

您应该自己转换数据格式。看起来 Brain.js 会自动检测 0 到 1 范围内的标准化数据，并且在训练之前不会应用任何修改。此处提供有关数据格式的文章

数据格式 使用 NeuralNetwork 进行训练 每个训练模式都应该有一个输入和一个输出，两者都可以是从 0 到 1 的数字数组或从 0 到 1 的数字哈希。

const net = new brain.NeuralNetwork();

net.train([
  { input: { r: 0.03, g: 0.7, b: 0.5 }, output: { black: 1 } },
  { input: { r: 0.16, g: 0.09, b: 0.2 }, output: { white: 1 } },
  { input: { r: 0.5, g: 0.5, b: 1.0 }, output: { white: 1 } },
]);

const output = net.run({ r: 1, g: 0.4, b: 0 }); // { white: 0.99, black: 0.002 }