javascript - Tensorflow.js使用节点保存模型

我想使用此功能从node.js保存经过训练的模型

async function tfModelExperiment(model) {
  try {
    let tsModelTraining = await model.save('file:///tmp/my-model-1');
  }
  catch (error) {
    console.log(error);
  }
}

但是在保存模型时会返回



我发现另一个人正在为这个问题而苦苦挣扎，但是通过解决这个问题

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

我已经尝试过将目录更改为我的主目录，但这并不能解决问题，也不能以sudo的身份运行它，我该怎么办？

软件
我正在将Ubuntu Ubuntu 18.04.1 LTS与npm一起安装的最新TensorFlow.js软件包(0.13.0)一起使用

编辑:

应该注意的是我尝试过

import * as tf from '@tensorflow/tfjs';
import '@tensorflow/tfjs-node';

如此处提供的(https://github.com/caisq/tfjs-node)，它返回



我尝试过:

const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');

返回与以前相同的UnhandledPromiseRejectionWarning错误

最佳答案

我现在在github的tfjs的帮助下工作了。

基本上，您只需要安装tfjs-node依赖项:
npm i @tensorflow/tfjs-node
然后，您只需要tfjs，它就可以工作。

const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');

const model = tf.sequential();
model.add(tf.layers.dense({units: 1, inputShape: [1]}));

model.save('file://./model-1a');