[英]Concatenate two layers
当我尝试连接两层的结果时,我遇到了一条错误消息。
def cnn_model_fn(learning_rate):
"""Model function for CNN."""
model1=Sequential()
# Convolutional Layer #1
model1.add(tf.keras.layers.Conv2D(
filters=20,
kernel_size=[10, 1],
kernel_initializer='he_uniform',
bias_initializer=keras.initializers.Constant(value=0),
padding="same",
activation=tf.nn.relu, input_shape=(410,1,3)))
model1.add(Flatten())
model2=Sequential()
model2.add(tf.keras.layers.Conv2D(
filters=20,
kernel_size=[10, 1],
kernel_initializer='he_uniform',
bias_initializer=keras.initializers.Constant(value=0),
padding="same",
activation=tf.nn.relu, input_shape=(410,1,3)))
model2.add(Flatten())
model4=Sequential()
model4.add(keras.layers.Concatenate(axis=-1)([model1, model2]))
optimizer = tf.train.AdamOptimizer(learning_rate)
model4.compile(loss='mean_squared_error',
optimizer=optimizer,
metrics=['mean_absolute_error', 'mean_squared_error'])
return model4
model4=cnn_model_fn(0.1)
model4.summary()
"/usr/local/lib/python3.6/site-packages/tensorflow/python/keras/layers/merge.py in build(self, input_shape) 377 # 纯粹用于形状验证。378 if not isinstance(input_shape, list ) 或 len(input_shape) < 2: --> 379 raise ValueError('A
Concatenate
layer should be called '380' on a list of at least 2 inputs') 381 if all([shape is None for shape in input_shape]) :ValueError:应在至少 2 个输入的列表上调用
Concatenate
层”
您正在尝试连接 2 个模型,但您想要的是连接 2 个层。 试试下面的代码。
from tensorflow.keras.models import Sequential, Model
from tensorflow.keras.layers import Flatten, Input
def cnn_model_fn(learning_rate):
"""Model function for CNN."""
input_layer=Input(shape=(410,1,3))
x1 = (tf.keras.layers.Conv2D(
filters=20,
kernel_size=[10, 1],
kernel_initializer='he_uniform',
bias_initializer=keras.initializers.Constant(value=0),
padding="same",
activation=tf.nn.relu ))(input_layer)
x1 = Flatten()(x1)
x2 = (tf.keras.layers.Conv2D(
filters=20,
kernel_size=[10, 1],
kernel_initializer='he_uniform',
bias_initializer=keras.initializers.Constant(value=0),
padding="same",
activation=tf.nn.relu))(input_layer)
x2 = Flatten()(x2)
x = (keras.layers.Concatenate(axis=-1)([x1,x2]))
model = Model(input_layer, x)
optimizer = tf.train.AdamOptimizer(learning_rate)
model.compile(loss='mean_squared_error',
optimizer=optimizer,
metrics=['mean_absolute_error', 'mean_squared_error'])
return model
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.