Когда я запускаю .predict
на моей модели, я возвращаю объект Tensor, но мне трудно найти все мои вероятности в одном месте.
Что-то, что я думал, будет в prediction.array()
, но это дало только одно предсказание. Любопытно, если модель не выводит результаты или я пропускаю метод.
Из руководств, которые я прошел, они просто делают;
const predictionResult = this._model.predict(inputTensor).dataSync();
const recognizedDigit = predictionResult.indexOf(Math.max(...predictionResult));
Самое странное для меня то, что когда я выполняю let result = prediction.dataSync()
, я получаю массив только с1 вероятность в этом, а не 5 я ожидаю.
Мой вопрос: какой метод я должен сделать, чтобы получить все вероятности для предсказания моей модели? (Просто нужно подтверждение, что модель возвращает неверную информацию или я что-то пропустил)