Guest User

Untitled

a guest
Jan 21st, 2019
83
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.55 KB | None | 0 0
  1. def lstm_net_with_dropout(input_shape, output_dim, dropout=0.15):
  2. inputs = keras.Input(shape=input_shape)
  3. x = keras.layers.LSTM(30, recurrent_dropout=dropout, return_sequences=True)(inputs, training=True)
  4. x = keras.layers.Dropout(dropout)(x, training=True)
  5. x = keras.layers.LSTM(30)(x, training=True)
  6. x = keras.layers.Dropout(dropout)(x, training=True)
  7. outputs = keras.layers.Dense(output_dim)(x)
  8. model = keras.Model(inputs, outputs)
  9. model.compile(loss='mean_squared_error', optimizer='adam', metrics=['mae', 'mse'])
  10. return model
Add Comment
Please, Sign In to add comment