Tensorflow.js tf.data.generator()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它还可以帮助开发人员使用JavaScript语言开发ML模型，并可以直接在浏览器或Node.js中使用ML。

tf.data.generator() 函数用于使用提供的 JavaScript 生成器创建数据集，该生成器生成每个元素。

用法:

tf.data.generator(generator)

参数：

generator:它是一个 JavaScript 生成器函数，它返回一个 JavaScript 迭代器。

返回值：它返回 tf.data.Dataset。

范例1：此示例说明如何从迭代器工厂创建数据集。

Javascript


// 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);
});

输出：

范例2：此示例说明如何从生成器创建数据集。

Javascript


// 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);
});

输出：

参考： https://js.tensorflow.org/api/3.6.0/#data.generator

相关用法

注：本文由纯净天空筛选整理自thacker_shahid大神的英文原创作品 Tensorflow.js tf.data.generator() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。