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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。