File size: 311 Bytes
2eee82e |
1 2 3 4 5 6 7 8 9 10 11 |
from tensorflow import keras
def build_lstm_model(seq_length):
model = keras.Sequential([
keras.layers.LSTM(50, return_sequences=True, input_shape=(seq_length, 1)),
keras.layers.LSTM(50),
keras.layers.Dense(1)
])
model.compile(optimizer='adam', loss='mse')
return model
|