将模型导入 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}`);
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}`);