张量流 predict_classes
Tensorflowjs predict_classes
我正在尝试使用 tensorflowjs 在 javascript 中导入 cnn classifier 并进行一些预测。有没有像 keras predict_classes 这样的方法，它只给我一个代表模型预测的 class 的数字？我可以使用正常的 model.predict，但我得到了一个张量，我正在努力迭代它以找到最高值。
没有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 reference 的文档。
我正在尝试使用 tensorflowjs 在 javascript 中导入 cnn classifier 并进行一些预测。有没有像 keras predict_classes 这样的方法，它只给我一个代表模型预测的 class 的数字？我可以使用正常的 model.predict，但我得到了一个张量，我正在努力迭代它以找到最高值。
没有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 reference 的文档。