Commit 1fa700f4 authored by Amelie Royer's avatar Amelie Royer

minor corrections

parent 59e547f7
......@@ -142,7 +142,7 @@ if __name__ == "__main__":
# write states
for i in xrange(n_choices):
c = cases[i]
current[c / (args.size - 1) + 1, c % (args.size - 1) + 1] = 60 if i < args.init else 120 if i < args.init + args.trap else 103 if i < args.init + args.trap + args.goal else 49
current[c // (args.size - 1) + 1, c % (args.size - 1) + 1] = 60 if i < args.init else 120 if i < args.init + args.trap else 103 if i < args.init + args.trap + args.goal else 49
# append new environment
str_maze = [[str(chr(x)) for x in line] for line in current]
mazes.append(str_maze)
......
......@@ -93,7 +93,7 @@ if __name__ == "__main__":
s2 = get_next_state_id(session[-1], a)
session.append(a)
session.append(s2)
f.write("%d\t%d\t%s\n" % (user, cluster + 1, ' '.join(str(x) for x in session) )) # Cluster 0 is the average transition probability for the MDP ==> not a user, no test sequences
f.write("%d\t%d\t%s\n" % (user, cluster, ' '.join(str(x) for x in session) )) # Cluster 0 is the average transition probability for the MDP ==> not a user, no test sequences
###### 4. Set rewards
print("\n\n\033[91m-----> Rewards generation\033[0m")
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment