Función Tensorflow.js tf.data.generator()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node. También ayuda a los desarrolladores a desarrollar modelos ML en lenguaje JavaScript y puede usar ML directamente en el navegador o en Node.js.

La función tf.data.generator() se utiliza para crear un conjunto de datos utilizando el generador de JavaScript proporcionado que produce cada elemento.

Sintaxis:

tf.data.generator(generator)

Parámetros:

• generador: es una función generadora de JavaScript que devuelve un iterador de JavaScript.

Valor devuelto: Devuelve tf.data.Dataset.

Ejemplo 1: este ejemplo ilustra cómo crear un conjunto de datos a partir de una fábrica de iteradores.

// Importing the tensorflow library
import * as tf from "@tensorflow/tfjs"
  
// Invoking .generator() function
let geek = tf.data.generator(function () {
  
    // Initializing variables used i.e.  
    // elements, index, result.
    let numbers = 4;
    let indices = 1;
    let outcome;
  
    // Creating and returning.next() function
    // in the format {value: TensorContainer,
    // done:boolean}.
    let repeator = {
        next: function () {
            if (indices <= numbers) {
                outcome =
                {
                    value: indices,
                    done: false
                };
                indices++;
                return outcome;
            }
            else {
                outcome =
                {
                    value: indices,
                    done: true
                };
                return outcome;
            }
        }
    };
    return repeator;
});
  
// Printing the result of returned promise
await geek.forEachAsync(function (geeks) {
    console.log(geeks);
});

Producción:

1
2
3
4

Ejemplo 2: Este ejemplo ilustra cómo crear un conjunto de datos a partir de un generador.

// Importing the tensorflow library
import * as tf from "@tensorflow/tfjs"
  
// Creating dataset from .generator() function 
let geek = tf.data.generator(function* () {
  
    // Initializing variables used i.e.  
    // data, and a.
    let data = 3;
    let a;
    for (i = 0; i <= data; ++i) {
        a = i;
        yield a;
    }
});
  
// Printing the result of returned promise
await geek.forEachAsync(function (geeks) {
    console.log(geeks);
});

Producción:

0
1
2
3

Referencia: https://js.tensorflow.org/api/3.6.0/#data.generator