Tensorflow js pred_classes

Question

Я пытаюсь импортировать классификатор cnn в javascript с тензорным потоком js и сделать некоторые прогнозы. Есть ли такой метод, как keras pred_classes, который дает мне только одно число, представляющее класс, прогнозируемый моделью? Я могу использовать обычный model.predict, но я получаю тензор и изо всех сил пытаюсь перебрать его, чтобы найти максимальное значение.

Answer 1

Нет такого метода, как predict_classes. Но вы можете сделать следующее.

const predictClasses = model.predict(input);
// this is considering that your predictClasses looks something like this [1,2,3]
const yourClass = predictClasses.argMax(-1).dataSync()[0]

Документ API для справки .