brozek/rltorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added missing parameter

Browse source
This commit is contained in:
Brandon Rozek 2019-02-03 21:57:27 -05:00
parent f1a52cf77b
commit 8c78f47c0c
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
rltorch/action_selector/EpsilonGreedySelector.py
View file

@ -9,7 +9,7 @@ class EpsilonGreedySelector(ArgMaxSelector):
# random_act is already implemented in ArgMaxSelector # random_act is already implemented in ArgMaxSelector
# best_act is already implemented in ArgMaxSelector # best_act is already implemented in ArgMaxSelector
def act(self, state): def act(self, state):
action = self.random_act() if np.random.rand() < self.epsilon else self.best_act() action = self.random_act() if np.random.rand() < self.epsilon else self.best_act(state)
if self.epsilon > self.epsilon_min: if self.epsilon > self.epsilon_min:
self.epsilon = self.epsilon * self.epsilon_decay self.epsilon = self.epsilon * self.epsilon_decay
return action return action