Así que tengo el modelo h5 Keras Tensorflow Deep Learning que acepta una matriz NumPy como entrada y da el resultado numérico deseado en forma de matriz NumPy (Python)
Ahora quiero hacer lo mismo en tensorflow.js, así que convertí el modelo, lo importé y usé la función model.predict (JavaScript)
const x = tf.tensor2d([company_name]) const output = model.predict(x) console.log(output)Y obtengo esto como salida
Tensor { kept: false, isDisposedInternal: false, shape: [ 1, 1 ], dtype: 'float32', size: 1, strides: [ 1 ], dataId: { id: 35 }, id: 52, rankType: '2', scopeId: 0 }Me gustaría obtener el valor real de tensor2d como 38.340986 que obtuve en la matriz NumPy. Traté de usar
model.predict(x.get(0,0));Pero devuelve un error de tipo.
Se agradece la ayuda.
Deberá descargar los tensores cuando use Tensorflow.js
const output = model.predict(x).dataSync()Los tensores son la estructura de datos central de TensorFlow.js. Son una generalización de vectores y matrices a dimensiones potencialmente más altas.
model.predict(np.array([value1], [value2])))