【问题标题】：Tensorflowjs predict_classesTensorflowjs predict_classes
【发布时间】：2020-11-02 23:41:57
【问题描述】：
我正在尝试使用 tensorflowjs 在 javascript 中导入一个 cnn 分类器并进行一些预测。有没有像 keras predict_classes 这样的方法，它只给我一个代表模型预测的类的数字？我可以使用正常的 model.predict，但我得到了一个张量，我正在努力迭代它以找到最高值。
【问题讨论】：
标签：
javascript
tensorflow
machine-learning
classification
tensorflow.js
【解决方案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]
reference 的 API 文档。