Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
def _build_compile_model(self):
model = Sequential()
model.add(Embedding(self._state_size, 10, input_length=1))
model.add(Reshape((10,)))
model.add(Dense(50, activation='relu'))
model.add(Dense(50, activation='relu'))
model.add(Dense(self._action_size, activation='linear'))
model.compile(loss='mse', optimizer=self._optimizer)
return model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment