如何修复“ model.predict不是函数”(tensorflow.js)?

时间:2019-10-31 12:58:45

标签: javascript tensorflow.js

我想测试以下错误代码:

'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用作教程。

1 个答案:

答案 0 :(得分: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);
});