projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[picoclvr.git]
/
escape.py
diff --git
a/escape.py
b/escape.py
index
7fcafaa
..
aa87cdd
100755
(executable)
--- a/
escape.py
+++ b/
escape.py
@@
-150,14
+150,16
@@
def episodes2str(states, actions, rewards, unicode=False, ansi_colors=False):
result = hline
for n in range(states.size(0)):
result = hline
for n in range(states.size(0)):
+
+ def state_symbol(v):
+ v = v.item()
+ return "?" if v < 0 or v >= len(symbols) else symbols[v]
+
for i in range(states.size(2)):
result += (
vert
+ vert.join(
for i in range(states.size(2)):
result += (
vert
+ vert.join(
- [
- "".join([symbols[v.item()] for v in row])
- for row in states[n, :, i]
- ]
+ ["".join([state_symbol(v) for v in row]) for row in states[n, :, i]]
)
+ vert
+ "\n"
)
+ vert
+ "\n"
@@
-166,8
+168,9
@@
def episodes2str(states, actions, rewards, unicode=False, ansi_colors=False):
result += (vert + thin_hori * states.size(-1)) * states.size(1) + vert + "\n"
def status_bar(a, r):
result += (vert + thin_hori * states.size(-1)) * states.size(1) + vert + "\n"
def status_bar(a, r):
- a = "ISNEW"[a.item()]
- r = "" if r == 0 else f"{r.item()}"
+ a = a.item()
+ a = "ISNEW"[a] if a >= 0 and a < 5 else "?"
+ r = "?" if r < -1 or r > 2 else ("" if r == 0 else f"{r.item()}")
return a + " " * (states.size(-1) - len(a) - len(r)) + r
result += (
return a + " " * (states.size(-1) - len(a) - len(r)) + r
result += (