+
81
-

tensorflow.js如何保存和加载model?

tensorflow.js能够在浏览器进行机器学习,确实很强大,但是如何保存和加载model?

网友回复

+
1
-

const saveResult = await model.save('localstorage://my-model-1');>

保存到浏览器localstorage中,key名为my-model-1

那么怎么加载?

const model = await tf.loadLayersModel('localstorage://my-model-1');

还可以在直接下载model

await model.save('downloads://my-model');

这会让浏览器下载模型文件至用户的机器上,并生成两个文件: 1. 一个名为 [my-model].json 的 JSON 文件,它包含了模型的拓扑结构和下面将要介绍的权重文件的引用。 2. 一个二进制文件,其中包含名为 [my-model].weights.bin 的权重值。 您可以更换 [my-model] 的名称以获得一个不同的名称的文件。 由于.json使用相对路径指向 .bin,所以两个文件需要被安放在同一个文件夹中。
我知道答案,我要回答