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


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


Tensorflow.js是Google開發的開源工具包,用於在瀏覽器或節點平台上執行機器學習模型和深度學習神經網絡。它還使開發人員能夠在 JavaScript 中創建機器學習模型,並直接在瀏覽器中或通過 Node.js 使用它們。

函數 tf.io.browserDownloads() 用於創建觸發從瀏覽器下載文件的 IOHandler。

用法:

tf.io.browserDownloads (fileNamePrefix?) 

參數:

  • fileNamePrefix (string):要下載的文件名稱的前綴。對於與 tf.Model 一起使用,fileNamePrefix 應為以下兩種類型之一:
    1. null 或未定義,在這種情況下將使用默認文件名。
    2. 單個字符串或單個字符串的數組。

返回:IO處理程序

示例 1:

Javascript


const model = tf.sequential(); 
model.add(tf.layers.dense( 
    { units: 1,  
    inputShape: [10],  
    activation: 'sigmoid' })); 
const res = await model.save('downloads://testModel'); 
console.log(res);

輸出:

{
 "modelArtifactsInfo": {
    "dateSaved": "2022-04-30T05:37:03.289Z",
    "modelTopologyType": "JSON",
    "modelTopologyBytes": 612,
    "weightSpecsBytes": 124,
    "weightDataBytes": 44
  }
}

示例 2:

Javascript


const model = tf.sequential(); 
model.add(tf.layers.dense( 
    { units: 10,  
    inputShape: [10],  
    activation: 'sigmoid'})); 
  
model.add(tf.layers.dense( 
    { units: 10,  
    inputShape: [10],  
    activation: 'sigmoid'})); 
  
model.add(tf.layers.dense( 
    { units: 1,  
    inputShape: [10],  
    activation: 'sigmoid'})); 
  
const res = await model.save('downloads://testModel'); 
console.log(res);

輸出:

{
   "modelArtifactsInfo": {
       "dateSaved": "2022-04-30T05:39:13.304Z",
       "modelTopologyType": "JSON",
       "modelTopologyBytes": 1570,
       "weightSpecsBytes": 374,
       "weightDataBytes": 924
   }
}

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



相關用法


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