预测位于何处？

Question

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

我认为应该在.predict中，但是仅产生了一个预测。好奇的是该模型没有输出结果还是我缺少一种方法。

从我看过的教程中，他们很简单；

prediction.array()

对我来说最奇怪的部分是，当我执行const predictionResult = this._model.predict(inputTensor).dataSync(); const recognizedDigit = predictionResult.indexOf(Math.max(...predictionResult)); 时，我得到的数组中只有1个概率，而不是我期望的5个概率。

我的问题是：我必须采取什么方法才能获得模型预测的所有概率？ （只需要确认模型会回馈不正确的信息，或者如果我错过了什么内容即可）

Answer 1

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