Как установить произвольный вес в слоях? - PullRequest
2 голосов
/ 28 мая 2020
• 1000 1009 *

Может ли кто-нибудь помочь мне, как set_weights в приведенном выше коде? Заранее спасибо.

1 Ответ

2 голосов
/ 28 мая 2020

Вы можете сделать это, используя метод set_weights.

Например, если вы хотите установить веса вашего LSTM Layer, его можно получить, используя model.layers[0], и если ваш Custom Weights , скажем, в массиве с именем my_weights_matrix, тогда вы можете установить Custom Weights на Первый уровень (LSTM), используя код, показанный ниже:

model.layers[0].set_weights([my_weights_matrix])

Если вы не хотите, чтобы ваши веса быть измененным во время обучения, тогда вам нужно заморозить этот слой с помощью кода model.layers[0].trainable = False.

Пожалуйста, дайте мне знать, если вы столкнетесь с какой-либо другой проблемой, и я буду рад вам помочь.

Надеюсь это поможет. Удачного обучения!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...