如何修复“model.predict 不是函数”（tensorflow.js）？

Question

我想测试以下代码是否缺少错误：

'use strict'

const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
const { createCanvas, createImageData } = require('canvas');

const canvas = createCanvas(800, 600);

async function load_model() {
    let m = await tf.loadLayersModel('path/to/model.json')
    return m;
}

let model = load_model();

model.then(function (res) {
    const example = tf.browser.fromPixels(canvas);
    const prediction = model.predict(example);
    console.log(prediction);
}, function (err) {
    console.log(err);
});

但我有一个） 它告诉TypeError: model.predict is not a function。为什么会这样？我用这个link 作为教程。

Answer 1

predict 是在那时返回的承诺上

model.then(function (res) {
    const example = tf.browser.fromPixels(canvas);
    const prediction = res.predict(example);
    console.log(prediction);
}, function (err) {
    console.log(err);
});