如何在TensorFlow 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”。无论何时进行预测，都可以将值重新映射到另一个数组。