将模型导入 TensorflowJs

Import models to TensorflowJs

const tf = require('@tensorflow/tfjs');
const model = tf.sequential();

const saveResult = model.save('file:/naive_bayes.pkl');

我想导入一个我在 Python 训练过的模型 tp TensorflowJs。但是我得到了这个错误:

(node:12312) UnhandledPromiseRejectionWarning: Error: Cannot find any save handlers for URL 'file:/naive_bayes.pkl'

我不明白,我正在关注它的文档。 https://www.tensorflow.org/js/guide/save_load?hl=vi

我把所有文件(模型、js)都放在同一个文件夹中。

这样使用

const modelDir = 'model' // model floder in same dir

const modelPath = `file://${modelDir}`;

model.save(modelPath);

或者简单地说,

const modelDir = 'model' 

model.save(`file://${modelDir}`);