Skip to content

Commit 28d2536

Browse files
committed
Changed argmax to idxmax.
1 parent 70a08d4 commit 28d2536

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

contents/2_Q_Learning_maze/RL_brain.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def choose_action(self, observation):
2424
# choose best action
2525
state_action = self.q_table.ix[observation, :]
2626
state_action = state_action.reindex(np.random.permutation(state_action.index)) # some actions have same value
27-
action = state_action.argmax()
27+
action = state_action.idxmax()
2828
else:
2929
# choose random action
3030
action = np.random.choice(self.actions)

0 commit comments

Comments
 (0)