如何在张量流 js 中控制输出范围

Question

我有这个模型：

const hidden = tf.layers.dense({
  units: 8,
  inputShape: [58, 8, 8],
  activation: 'sigmoid'
});
const output = tf.layers.dense({
  units: 1,
  activation: 'softmax'
});
var model = tf.sequential({
  layers: [
  hidden,
  output
  ]
});

现在我希望输出在 -1 和 1 范围内，我该如何实现？

Answer 1

激活函数是决定输出的因素。"sigmoid"适用于 0 到 1 输出，"tanh"适用于 -1 到 1，但是您选择使用通常用于多类分类的"softmax"。无论何时进行预测，都可以将值重新映射到另一个数组。