具有 softmax 输出层的模型,该层不输出概率,只输出 1 和 0
作者: croraf创建于 2024年5月6日更新于 2026年7月8日
标签comp:node.jscomp:modelstype:othersstat:contributions welcome
我正在测试 node-js 示例中提供的模型 https://GitHub.com/tensorflow/tfjs-examples/blob/master/mnist-node/model.js,用于将数字的图像分类为 0 到 9 这几个类别。这是一个 `tf.sequential()` 模型,其最后一层为 `model.add(tf.layers.dense({ units: 10, activation: "softmax" }));` 因此预测结果应该是每个类别的概率,但并非如此。相反,只输出了 0 和 1(以 one-hot 方式),如下面的日志所示。
内容来源: tensorflow/tfjs