Tensorflowjs predict_classes

Question

我正在嘗試使用 tensorflowjs 在 javascript 中導入一個 cnn 分類器並進行一些預測。 有沒有像 keras predict_classes 這樣的方法，它只給我一個代表 class 的數字 model 正在預測？ 我可以使用普通的 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 文檔供參考。