python - 为 Keras 中的顺序节点提供确切名称

Question

我试图在 Keras 模型中命名一个节点，但 Keras 将附加信息附加到层名称。

例如，当您创建一个名为“sm_output”的图层时:

model.add(keras.layers.Dense(output_size, activation='softmax', name="sm_output"))

你最终得到了被调用的 softmax 输出

sm_output/Softmax

如何命名模型的最终输出？我需要它符合标准约定，因为此模型与其他模型互换，如果不相同，调用代码将无法找到输出层。

Answer 1

@Lukeyb 您可以使用 model.layers[-1].name 访问您最初设置的最后一层名称。您可以更改层的索引以访问模型中其他层的名称

请检查下面的示例，其中访问了最后一层的名称。

import tensorflow as tf

model1 = tf.keras.models.Sequential([
  tf.keras.layers.Dense(128, activation='relu',input_shape=(None, None, 784,)),
  tf.keras.layers.Dense(256, activation='relu'),
  tf.keras.layers.Dense(10, activation='softmax',name='last_layer')
])

model1.layers[-1].name

输出如下

'last_layer'