projects
/
picoclvr.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (from parent 1:
a3f5378
)
Update.
author
François Fleuret
<francois@fleuret.org>
Sun, 18 Feb 2024 21:40:29 +0000
(22:40 +0100)
committer
François Fleuret
<francois@fleuret.org>
Sun, 18 Feb 2024 21:40:29 +0000
(22:40 +0100)
tasks.py
patch
|
blob
|
history
diff --git
a/tasks.py
b/tasks.py
index
00b7a49
..
181ac44
100755
(executable)
--- a/
tasks.py
+++ b/
tasks.py
@@
-136,12
+136,9
@@
class TaskFromFile(Task):
pairs.append((sequence, pred_mask))
symbols = ["#"] + list(set("".join([x[0] for x in pairs])) - set(["#"]))
pairs.append((sequence, pred_mask))
symbols = ["#"] + list(set("".join([x[0] for x in pairs])) - set(["#"]))
- print("SANITY", symbols)
self.char2id = dict([(c, n) for n, c in enumerate(symbols)])
self.id2char = dict([(n, c) for c, n in self.char2id.items()])
self.char2id = dict([(c, n) for n, c in enumerate(symbols)])
self.id2char = dict([(n, c) for c, n in self.char2id.items()])
- print(self.char2id)
-
self.train_input, self.train_pred_masks = self.tensorize(
pairs[:nb_train_samples]
)
self.train_input, self.train_pred_masks = self.tensorize(
pairs[:nb_train_samples]
)
@@
-163,7
+160,6
@@
class TaskFromFile(Task):
return len(self.char2id)
def tensor2str(self, t):
return len(self.char2id)
def tensor2str(self, t):
- print(f"{type(t)=}")
return ["".join([self.id2char[x.item()] for x in s]) for s in t]
def produce_results(
return ["".join([self.id2char[x.item()] for x in s]) for s in t]
def produce_results(