我有这样一个模型:
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之间,我该如何实现呢?
回答:
激活函数决定了输出的范围。’sigmoid’适合0到1的输出,’tanh’适合-1到1的输出,不过你选择使用了’softmax’,这通常用于多类分类。不管怎样,当你进行预测时,你可以将值重新映射到另一个数组中。