Función Tensorflow.js tf.inTopKAsync()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático, así como redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.

La función .inTopKAsync() se utiliza para verificar si los objetivos indicados se encuentran en las K predicciones principales dadas .

Sintaxis:  

tf.inTopKAsync(predictions, targets, k?)

Parámetros:  

• Predicciones: es la entrada de 2-D o tensor superior indicada cuyo último tamaño no es menor que k y puede ser de tipo tf.Tensor, TypedArray o Array.
• objetivos: es la entrada 1-D o tensor superior indicada y puede ser de tipo tf.Tensor, TypedArray o Array.
• k: Es el número alternativo de elementos superiores que se van a considerar para calcular la precisión. El valor por defecto es 1 y es de tipo número.

Valor devuelto: Devuelve el objeto Promise tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining predictions and targets
const pred = tf.tensor2d([
    [11, 22, 33, 55], 
    [33, 66, 22, -11]
]);
  
const targ = tf.tensor1d([1, 1]);
  
// Calling tf.inTopKAsync() method
const res = await tf.inTopKAsync(pred, targ);
  
// Printing output
res.print();

Producción:

Tensor
    [false, true]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.inTopKAsync() method with 
// all its parameters
const res = await tf.inTopKAsync(
    tf.tensor2d([[1.1, 2.2], [3.3, 6.6]]), 
    tf.tensor1d([0, 1]), 2);
  
// Printing output
res.print();

Producción:

Tensor
    [true, true]

Referencia: https://js.tensorflow.org/api/latest/#inTopKAsync