从tensorflow.js神经网络获取权重

2024-02-05

我有这个顺序模型:

this.model = tf.sequential()

this.model.add(tf.layers.dense({units : 16, useBias : true, inputDim : 7})) // input
this.model.add(tf.layers.dense({units : 16, useBias : true, activation: 'sigmoid'})) // hidden
this.model.add(tf.layers.dense({units : 3, useBias : true, activation: 'sigmoid'})) // hidden 2

我检查了tensorflow.js的API,但是没有关于获取神经网络的权重(内核)的信息。那么,我如何获得权重,然后更改它们以应用新的权重?(用于无监督学习)


这是打印所有权重的简单方法:

for (let i = 0; i < model.getWeights().length; i++) {
    console.log(model.getWeights()[i].dataSync());
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

从tensorflow.js神经网络获取权重 的相关文章

随机推荐