(self, state_size, action_size)
| 258 | |
| 259 | class DQNAgent(object): |
| 260 | def __init__(self, state_size, action_size): |
| 261 | self.state_size = state_size |
| 262 | self.action_size = action_size |
| 263 | self.memory = ReplayBuffer(state_size, action_size, size=500) |
| 264 | self.gamma = 0.95 # discount rate |
| 265 | self.epsilon = 1.0 # exploration rate |
| 266 | self.epsilon_min = 0.01 |
| 267 | self.epsilon_decay = 0.995 |
| 268 | self.model = mlp(state_size, action_size) |
| 269 | |
| 270 | |
| 271 | def update_replay_memory(self, state, action, reward, next_state, done): |
nothing calls this directly
no test coverage detected