functional-programming - 在函数式 API 中简化 Keras LSTM 模型

Question

我有以下使用功能 API 的 Keras LSTM 模型:

model = Sequential()
model.add(Lambda(lambda x: x,input_shape=(timestep,n_feature)))
output = model.output
output = LSTM(8)(output)
output = Dense(2)(output)

inputTensor = model.input
myModel = Model([inputTensor], output)
myModel.compile(loss='mean_squared_error', optimizer='adam')

myModel.fit([trainX], trainY, epochs=100, batch_size=1, verbose=2, validation_split = 0.1)

该模型工作正常，但我认为我的架构中有多余的语法。例如，Lambda 层仅用于定义 input_shape，也许可以将其删除？上面的代码可以简化/清理吗(我想继续使用函数式API)？谢谢!

Answer 1

您可以使用功能 API 编写模型，如下所示 -

x=Input(shape=(timestep,n_feature))
model=LSTM(8)(x)
model=Dense(2)(model)

myModel=Model(x,model)