使用 TensorFlow.js 更改权重

TensorFlow.js 是一个开源的 JavaScript 机器学习库，可用于在浏览器和 Node.js 中训练和部署机器学习模型。本文将介绍如何在 TensorFlow.js 中更改模型的权重。

加载模型

要更改模型的权重，首先需要加载模型。可以使用 tf.loadLayersModel() 方法从本地文件或 URL 加载模型，或者使用预先训练的模型。例如，以下代码加载了预训练的 MobileNet 模型：

const model = await tf.loadLayersModel('https://tfhub.dev/google/tfjs-model/imagenet/mobilenet_v2_100_224/feature_vector/3');

查看和更改权重

一旦加载了模型，可以使用 model.weights 属性查看和更改所有权重。此属性返回一个张量数组，其中每个张量对应一个层的权重。

例如，以下代码将显示模型中第一个卷积层的权重：

console.log(model.weights[0].dataSync());

要更改权重，可以使用 tf.tensor() 方法创建一个新的张量，并将其分配给权重的 .val 属性。例如，以下代码将更改第一个卷积层的权重：

const newWeights = tf.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]]);
model.weights[0].val.assign(newWeights);
console.log(model.weights[0].dataSync());

保存模型

更改权重后，可以使用 model.save() 方法将其保存到本地文件或 URL。例如，以下代码将模型保存到本地文件：

await model.save('file://my_model');

总结

在 TensorFlow.js 中更改模型权重可以使用 model.weights 属性访问和更改所有权重。更改权重后，可以使用 model.save() 方法将其保存到本地文件或 URL。