Skip to content

Instantly share code, notes, and snippets.

@sol0invictus

sol0invictus/ddpg-model.py

Created May 29, 2020 17:42
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 sol0invictus/fcb039058ad14501a48a87bb0f23ac38 to your computer and use it in GitHub Desktop.
Save sol0invictus/fcb039058ad14501a48a87bb0f23ac38 to your computer and use it in GitHub Desktop.
Download ZIP
DDPG Blog - Model generation function
Raw
ddpg-model.py
def ANN2(input_shape,layer_sizes, hidden_activation='relu', output_activation=None):
model = tf.keras.Sequential()
model.add(tf.keras.layers.Input(shape=input_shape))
for h in layer_sizes[:-1]:
x = model.add(tf.keras.layers.Dense(units=h, activation='relu'))
model.add(tf.keras.layers.Dense(units=layer_sizes[-1], activation=output_activation))
return model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment