| 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 | |