如何在對應於 numpy 數組的 tensorflow.js 模型中獲取 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
}

我想獲得我在 NumPy 數組中獲得的 tensor2d 的實際值，如 38.340986。 我嘗試使用

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

但它返回一個類型錯誤

幫助表示贊賞。

Answer 1

使用Tensorflow.js時需要下載張量

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

Answer 2

張量是 TensorFlow.js 的核心數據結構，它們是向量和矩陣到潛在更高維度的泛化。

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

Answer 3

我用這個解決了：

    console.log(model.predict(tf.tensor2d([i],[1, 1])).dataSync()[0]);