将模型导入 TensorFlowJs

Question

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）放在同一个文件夹中。

Answer 1

这样使用

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

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

model.save(modelPath);

或者简单地说，

const modelDir = 'model' 

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