seq[:, 1:] = -m[:, :-1] + (1 - m[:, :-1]) * seq[:, 1:]
-def seq_to_str(seq, recorded_stack_counts=None):
+def seq_to_str(seq, show_stack_nb=True,recorded_stack_counts=None):
assert seq.size(0) % 2 == 0
s = ""
for t in range(seq.size(0) // 2):
- op = seq[2 * t]
- op = f"POP_{op//2}" if op % 2 == 1 else f"PSH_{op//2}"
+ n_op = seq[2 * t]
+ op = f"POP" if n_op % 2 == 1 else f"PSH"
+ if show_stack_nb: op+=f"_{n_op//2}"
if seq[2 * t + 1] == -1:
val = "?"
else:
nb=nb, nb_steps=nb_steps, nb_stacks=nb_stacks, nb_values=nb_values
)
+ print("-- TRAIN -----------------------------")
+
for n in range(min(10, seq.size(0))):
# print(seq_to_str(seq[n], recorded_stack_counts[n]))
- print(seq_to_str(seq[n]))
+ print(seq_to_str(seq[n],show_stack_nb=nb_stacks>1))
- print("--------------------------------------")
+ print("-- TEST ------------------------------")
remove_poped_values(seq, nb_stacks)
for n in range(min(10, seq.size(0))):
- print(seq_to_str(seq[n]))
+ print(seq_to_str(seq[n],show_stack_nb=nb_stacks>1))