在Tensorflow.js中使用loadLayersModel加载Keras模型时出现”仅支持绝对URL”错误

我想在NodeJS服务器中从本地文件加载Keras模型到Tensorflow.js中,但出现了以下错误:“仅支持绝对URL”。

let tf = require("@tensorflow/tfjs");(async () => {    try    {        const model = await tf.loadLayersModel("/path/to/model.json");    }    catch(error)    {        console.error(error);    }})();

loadLayersModel是否还不支持本地文件?

谢谢!


回答:

Tensorflow的文档指出,您应该使用file://关键字直接访问文件系统,例如

tf.loadLayersModel("file://path/to/model.json");

模型的路径是相对于您当前调用函数的文件夹。例如,如果上述函数位于/a/b/c文件夹中的文件中,而模型位于/a/d/model.json,那么正确的路径是“file://../../d/model.json”。

此外,需要使用require(‘@tensorflow/tfjs-node’),否则会抛出以下错误:“仅支持HTTP(s)协议”。

完整的工作示例:

const tf = require('@tensorflow/tfjs');require('@tensorflow/tfjs-node');(async () => {    try    {        const model = await tf.loadLayersModel('file://relative/path/to/model.json');    }    catch(error)    {        console.error(error);    }})();

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注