Tensorflow.js tf.io.http()函数
Tensorflow.js 是由谷歌开发的开源库,用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。
.io.http()函数用于生成IOHandler子集,该子集将模型工件传输到 HTTP 服务器。此外,多部分或表单数据mime 类型的 HTTP 请求应传输到路径URL。其中,表单数据包含描述模型拓扑和/或模型权重的工件。
句法:
tf.io.http(path, loadOptions?)
参数:
- 路径:模型的指定 URL 路径。此外,它可以是完整的 HTTP 路径,即“http://localhost:8000/model-upload”或类似的路径,即“./model-upload”。它是字符串类型。
- loadOptions:用于加载的指定配置。它是可选的,属于LoadOptions类型。它包括以下字段:
- weightPathPrefix:权重文件的路径可选前缀。此外,默认情况下 this 的值是从路径参数评估的。
- fetchFunc:指定的可选自定义获取函数。
- 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);
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);
输出:
{
"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