Advertisement
SEnergy

Untitled

Nov 21st, 2018
101
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.47 KB | None | 0 0
  1. import keras;
  2. import numpy as np;
  3.  
  4. model = keras.models.Sequential();
  5.  
  6. model.add(keras.layers.LSTM(7, input_shape=(5, 1), stateful=True, batch_size=6, name="input"));
  7. model.add(keras.layers.Dense(4, activation="relu", name="hidden1"));
  8. model.add(keras.layers.Dense(2, activation="softmax", name="output"));
  9. model.compile(optimizer="rmsprop", loss="binary_crossentropy", metrics=["accuracy"]);
  10.  
  11. keras.utils.plot_model(model, show_shapes=True);
  12. model.save("model.h5");
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement