Skip to content

Commit dc10681

Browse files
committedDec 14, 2016
add RL example2
1 parent 0740418 commit dc10681

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed
 

‎RL/example2/maze.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def get_reward_and_next_state(s, a):
111111
s_ = 'terminal'
112112
else:
113113
reward = 0
114-
return s_, reward
114+
return reward, s_
115115

116116

117117
# update loop
@@ -121,9 +121,9 @@ def update():
121121
is_terminated = False
122122
while not is_terminated:
123123
A = QLearn.choose_action(str(S))
124-
S_, R = get_reward_and_next_state(S, A) # take action & get next state and reward
124+
R, S_ = get_reward_and_next_state(S, A) # take action & get reward and next state
125125
QLearn.update_table(str(S), A, R, str(S_))
126-
S = S_
126+
S = S_ # state will become next state
127127
window.update()
128128
time.sleep(0.1)
129129
if S_ == 'terminal':

0 commit comments

Comments
 (0)
Please sign in to comment.