Skip to content

Instantly share code, notes, and snippets.

@vivekpadia70
Created February 20, 2020 08:58
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 vivekpadia70/5df43127f433a0fe589b2737534c7ce3 to your computer and use it in GitHub Desktop.
Save vivekpadia70/5df43127f433a0fe589b2737534c7ce3 to your computer and use it in GitHub Desktop.
model = Sequential()
model.add(Flatten(input_shape=(1,) + env.observation_space.shape))
model.add(Dense(16))
model.add(Activation('relu'))
model.add(Dense(nb_actions))
model.add(Activation('linear'))
print(model.summary())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment