當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Tensorflow.js tf.io.http()用法及代碼示例


Tensorflow.js是由Google開發的開源庫,用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.io.http() 函數用於生成將模型工件傳輸到 HTTP 服務器的 IOHandler 子集。此外,一個 HTTP 請求multi-part或者form-datamime 類型應傳輸到小路網址。其中,form-data 包含描述模型拓撲和/或模型權重的工件。

用法:

tf.io.http(path, loadOptions?)

參數:

  • path:模型的指定 URL 路徑。此外,它可以是完整的 HTTP 路徑,即“http://localhost:8000/model-upload”或類似的路徑,即“./model-upload”。它是字符串類型。
  • loadOptions:用於加載目的的所述配置。它是可選的,屬於 LoadOptions 類型。它包括以下字段:
    1. weightPathPrefix:指定的權重文件路徑的可選前綴。此外,默認情況下 this 的值是從路徑參數中評估的。
    2. fetchFunc:所述的可選自定義fetch 函數。
    3. onProgress:聲明的可選進度回調函數,它在加載完成之前定期釋放。

返回值:它返回 IOHandler。



範例1:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling io.http() method
const result = tf.io.http('https://js.tensorflow.org/api/latest/#io.http');
  
// Printing output
console.log(result);

輸出:

{
  "DEFAULT_METHOD":"POST",
  "path":"https://js.tensorflow.org/api/latest/#io.http",
  "requestInit":{}
}

範例2:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating model
const model = tf.sequential();
  
// Adding layer to the model
model.add(
     tf.layers.dense({units:2, inputShape:[10]}));
  
// Calling io.http() method within 
// save() method
const result = await model.save(tf.io.http(
     'https://js.tensorflow.org/api/latest/#io.http'));
  
// Printing output
console.log(result);

輸出:

{
  "modelArtifactsInfo":{
    "dateSaved":"2021-08-26T08:43:49.648Z",
    "modelTopologyType":"JSON",
    "modelTopologyBytes":611,
    "weightSpecsBytes":124,
    "weightDataBytes":88
  },
  "responses":[
    {}
  ]
}

參考: https://js.tensorflow.org/api/latest/#io.http




相關用法


注:本文由純淨天空篩選整理自nidhi1352singh大神的英文原創作品 Tensorflow.js tf.io.http() Function。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。