INNER CODE UNIT · Python
epsilon_decay
rasbt/machine-learning-book · ch19/cartpole/main.py:37
epsilon_decay=0.995, learning_rate=1e-3,
max_memory_size=2000):
self.env = env
self.state_size = env.observation_space.shape[0]
self.action_size = env.action_space.n
self.memory = deque(maxlen=max_memory_size)
self.gamma = discount_factor
self.epsilon = epsilon_greedy
self.epsilon_min = epsilon_min
self.epsilon_decay = epsilon_decay
self.lr = learning_rate
self._build_nn_model()
def _build_nn_model(self):
self.model = nn.Sequential(nn.Linear(self.state_size, 256),
nn.ReLU(),