- nb, seq_len, nb_stacks, nb_values = 3, 10, 1, 5
- result = generate(nb=nb, seq_len=seq_len, nb_stacks=nb_stacks, nb_values=nb_values)
- for n in range(result.size(0)):
- print(seq_to_str(result[n]))
+ nb, nb_steps, nb_stacks, nb_values = 150000, 10, 1, 5
+ seq, recorded_stack_counts = generate_sequences(
+ 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],show_stack_nb=nb_stacks>1))
+
+ print("-- TEST ------------------------------")
+
+ remove_poped_values(seq, nb_stacks)
+
+ for n in range(min(10, seq.size(0))):
+ print(seq_to_str(seq[n],show_stack_nb=nb_stacks>1))