我无法使用@ tensorflow / tfjs从file：//加载模型

Question

我正在构建一个npm软件包，该软件包会加载预训练的张量流模型并进行一些预测。

当我对tf.loadLayersModel()做tf = require('@tensorflow/tfjs-node')时，我没有问题。不幸的是，该库在安装时似乎有问题。 我尝试改用tf = require('@tensorflow/tfjs)。虽然，但出现以下错误：

UnhandledPromiseRejectionWarning：错误：没有获取polyfill的Web浏览器外部不支持browserHTTPRequest。

这是我的代码：

 // Load ML model
  tf.loadLayersModel("file://" + __dirname + "mymodel/model.json").then(
    model => {
      }
    }
  );

有没有办法解决这个问题？

Answer 1

  

UnhandledPromiseRejectionWarning：错误：没有获取polyfill的Web浏览器外部不支持browserHTTPRequest。

如果您使用@tensorflow/tfjs而不是@tensorflow/tfjs-node，则需要使用polyfill进行提取

global.fetch = require('node-fetch');

Answer 2

在使用Node.js时，最好使用@tensorflow/tfjs-node。 只是不需要@tensorflow/tfjs，一切都会按预期进行。

但是，如果您真的只想使用@tensorflow/tfjs，则只需添加tfjs-node-save

npm install --save tfjs-node-save

并在脚本中使用

require('tfjs-node-save');