尝试学习 tensorflow.js 但需要一个更简单的示例，例如 brain.js

Question

brain.js 的示例让我真正了解了该软件 - 以及机器学习的各个方面。

现在我正在努力学习，很难复制相同类型的代码 tensorflow.js。

比如TensorFlow相当于下面的脑码是什么？

var 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 }}]);

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

Answer 1

这将是您提供的示例的一种简化版本：

const net = tf.sequential();
net.add(tf.layers.dense({
  units: 2,
  inputShape: [3],
  activation: 'sigmoid'
}));
net.compile({
  loss: 'meanSquaredError',
  optimizer: 'sgd'
});

const xs = tf.tensor2d([
  [0.03, 0.7, 0.5],
  [0.16, 0.09, 0.2],
  [0.5, 0.5, 1.0]
]);
const ys = tf.tensor2d([
  [1, 0],
  [0, 1],
  [0, 1]
]);

net.fit(xs, ys).then(() => {
  const xPredict = tf.tensor2d([
    [1.0, 0.4, 0.0]
  ]);

  const prediction = net.predict(xPredict);
  prediction.print();
});

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.11.6">
</script>

但是正如您所见，您必须更具体地了解神经网络的实际作用。我只是使用了您可以使用的基本选项。正如您所看到的，网络因此性能很差，但我希望您了解它是如何工作的。所以让我试着解释一下这段代码的作用：

const 通常用于 tf.tensor，因为它们存在于 GPU 上，因此无论如何都无法更改。

tf.sequential() 创建一个空的前馈网络。（如果您不知道那是什么，请先尝试了解神经网络，但不要实施）

tf.layers.dense() 创建一个全连接层。

units:2 定义层的输出形状。在这种情况下，一个向量有两个值。

inputShape: [3] 可以在任何非第一层上忽略，因为它可以由前一层推断并定义输入张量的形状

activation: 'sigmoid' 是应用于层的 return 值的激活函数，高度依赖于您要解决的问题。

.compile() 使用给定的选项编译网络并且也非常可定制

xs和ys是训练数据集。注意：它们有一个额外的维度来表示多个x-y-pair，所以它们可以批量传递给训练函数。

.fit()是训练方式，训练网络内部的权重。注意：这是一个异步函数，所以你必须等它完成才能使用模型。

xPredict是测试数据，也是比网络的return形状高一维。

.predict() 根据给定的输入预测网络的输出。

.print() 在控制台输出张量。（如果太大，就会被裁剪）

我强烈建议您先了解更多有关神经网络的知识，然后再尝试（通过复制）实现它们，因为它们会变得非常复杂和令人困惑。然后你可以阅读文档，看看有什么可能。

尝试学习 tensorflow.js 但需要一个更简单的示例，例如 brain.js

Trying to learn tensorflow.js but need a simpler example like brain.js

javascript

node.js

brain.js

tensorflow.js