projects
/
mygptrnn.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[mygptrnn.git]
/
problems.py
diff --git
a/problems.py
b/problems.py
index
9e368c2
..
3cdd374
100755
(executable)
--- a/
problems.py
+++ b/
problems.py
@@
-149,7
+149,13
@@
class ProblemMemory(Problem):
return sequences, ar_mask
def seq2str(self, seq):
return sequences, ar_mask
def seq2str(self, seq):
- return "".join(self.token_string[x.item()] for x in seq)
+ def decode(x):
+ if x < len(self.token_string):
+ return self.token_string[x]
+ else:
+ return "?"
+
+ return "".join(decode(x.item()) for x in seq)
class ProblemTwoTargets(Problem):
class ProblemTwoTargets(Problem):