Где находятся прогнозы? - PullRequest
2 голосов
/ 02 декабря 2019

Когда я запускаю .predict на моей модели, я возвращаю объект Tensor, но мне трудно найти все мои вероятности в одном месте.

Что-то, что я думал, будет в prediction.array(), но это дало только одно предсказание. Любопытно, если модель не выводит результаты или я пропускаю метод.

Из руководств, которые я прошел, они просто делают;

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

Самое странное для меня то, что когда я выполняю let result = prediction.dataSync(), я получаю массив только с1 вероятность в этом, а не 5 я ожидаю.

Мой вопрос: какой метод я должен сделать, чтобы получить все вероятности для предсказания моей модели? (Просто нужно подтверждение, что модель возвращает неверную информацию или я что-то пропустил)

...