Skip to content

Instantly share code, notes, and snippets.

@ranpelta
Created July 18, 2020 07:53
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ranpelta/81613aa15c4a697b234b83052726e1bc to your computer and use it in GitHub Desktop.
Save ranpelta/81613aa15c4a697b234b83052726e1bc to your computer and use it in GitHub Desktop.
model = Sequential()
model.add(LSTM(150, activation='relu', input_shape=(n_input, n_features)))
model.add(Dense(1))
model.compile(optimizer='adam', loss='mse')
model.summary()
>>>
Model: "sequential_1"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
lstm_1 (LSTM) (None, 150) 92400
_________________________________________________________________
dense_1 (Dense) (None, 1) 151
=================================================================
Total params: 92,551
Trainable params: 92,551
Non-trainable params: 0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment