Получение весов из нейронной сети tenorflow.js - PullRequest
0 голосов
/ 30 апреля 2018

У меня есть эта последовательная модель:

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

Я проверил API для tenorsflow.js, но нет ничего о получении весов (ядер) нейронной сети. Итак, как я могу получить веса, а затем изменить их, чтобы применить новые веса (для обучения без учителя)

1 Ответ

0 голосов
/ 24 мая 2018

Кажется, что, возможно, есть более простой и чистый способ делать то, что вы хотите, но независимо от этого:

Вызов this.model.getWeights() даст вам массив переменных, которые соответствуют весам и уклонам слоя. Вызов data() для любого из этих элементов массива вернет обещание, которое вы можете решить, чтобы получить весовые коэффициенты.

Я не пробовал вручную устанавливать веса, но есть метод this.model.setWeights().

Гудлак.

...