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：

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：

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