我正在尝试使用tensorflow.keras创建一个完全连接的模型,以下是我的代码
from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input, Dense, Flatten
def load_model(input_shape):
input = Input(shape = input_shape)
dense_shape = input_shape[0]
x = Flatten()(input)
x = Dense(dense_shape, activation='relu')(x)
x = Dense(dense_shape, activation='relu')(x)
x = Dense(dense_shape, activation='relu')(x)
x = Dense(dense_shape, activation='relu')(x)
x = Dense(dense_shape, activation='relu')(x)
output = Dense(10, activation='softmax')
model = Model(input , output)
model.summary()
return model但是当我调用模型时
model = load_model((120,))我有这个错误
'Dense' object has no attribute 'op'我该如何解决这个问题呢?
发布于 2020-04-07 23:03:23
您的输出层后面缺少(x)。试一试
output = Dense(10 , activation = 'softmax')(x)https://stackoverflow.com/questions/61083004
复制相似问题