如何在 tensorflow.js 模型中获取对应于 numpy 数组的 tensor2d 的值

Question

所以我有 h5 Keras Tensorflow 深度学习模型，它接受 NumPy 数组作为输入并以 NumPy 数组的形式给出所需的数值结果 (Python)

现在我想在 tensorflow.js 中做同样的事情所以我转换模型并导入它并使用 model.predict 函数 (JavaScript)

    const x = tf.tensor2d([company_name])
    const output = model.predict(x)
    console.log(output)

我得到这个作为输出

Tensor {
  kept: false,
  isDisposedInternal: false,
  shape: [ 1, 1 ],
  dtype: 'float32',
  size: 1,
  strides: [ 1 ],
  dataId: { id: 35 },
  id: 52,
  rankType: '2',
  scopeId: 0
}

我想得到 tensor2d 的实际值，比如我在 NumPy 数组中得到的 38.340986。我尝试使用

model.predict(x.get(0,0));

但它 returns 类型错误

感谢帮助。

Answer 1

张量是 TensorFlow.js 的核心数据结构，它们是向量和矩阵向潜在更高维度的泛化。

model.predict(np.array([value1], [value2])))

Answer 2

使用Tensorflow.js

时需要下载张量

const output = model.predict(x).dataSync()

如何在 tensorflow.js 模型中获取对应于 numpy 数组的 tensor2d 的值

How to get value of tensor2d in tensorflow.js model which corresponds to a numpy array

javascript

node.js

tensorflow

deep-learning

tensorflow.js