Разделение веса в Керасе - PullRequest
Новая
       6

Разделение веса в Керасе

0 голосов
/ 28 февраля 2019

Я пытаюсь использовать одинаковый вес для двух слоев, которые названы dense3 в коде ниже.Я знаю, что использование dense3 в качестве имени недопустимо, поскольку слои должны иметь уникальные имена.

Как лучше всего указывать разные слои в Керасе одинакового веса? 

model = keras.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(128, activation=tf.nn.relu,name='dense_1'),
    keras.layers.Dense(10, activation=tf.nn.relu, name='dense_2'),
    keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' ),
    keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' ),
    keras.layers.Dense(10, activation=tf.nn.softmax)
])

1 Ответ

0 голосов
/ 28 февраля 2019 
sharedLayer = keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' )

model = keras.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(128, activation=tf.nn.relu,name='dense_1'),
    keras.layers.Dense(10, activation=tf.nn.relu, name='dense_2'),
    sharedLayer,
    sharedLayer,
    keras.layers.Dense(10, activation=tf.nn.softmax)
])
...