Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def lstm_net_with_dropout(input_shape, output_dim, dropout=0.15):
- inputs = keras.Input(shape=input_shape)
- x = keras.layers.LSTM(30, recurrent_dropout=dropout, return_sequences=True)(inputs, training=True)
- x = keras.layers.Dropout(dropout)(x, training=True)
- x = keras.layers.LSTM(30)(x, training=True)
- x = keras.layers.Dropout(dropout)(x, training=True)
- outputs = keras.layers.Dense(output_dim)(x)
- model = keras.Model(inputs, outputs)
- model.compile(loss='mean_squared_error', optimizer='adam', metrics=['mae', 'mse'])
- return model
Add Comment
Please, Sign In to add comment