javascript - 预测位于何处？

Question

当我在我的模型上运行 .predict 时，我得到了一个 Tensor 对象，但我很难在一个地方找到我所有的概率。

我认为 prediction.array() 中会有一些东西，但只产生了一个预测。好奇模型是否没有输出结果或者我是否缺少方法。

从我学习过的教程来看，他们就是这么做的；

const predictionResult =  this._model.predict(inputTensor).dataSync();
const recognizedDigit = predictionResult.indexOf(Math.max(...predictionResult));

对我来说最奇怪的部分是，当我执行 let result = prediction.dataSync() 时，我得到一个数组，其中只有 1 个概率，而不是我期望的 5 个。

我的问题是:我必须采用什么方法来获得模型预测的所有概率？ (只需要确认模型返回不正确的信息或者我是否遗漏了什么)

Answer 1

如果您只有一个预测，那么您的输出层中似乎只有一个神经元。因此，请检查您的模型并确保您的输出层具有正确的形状。