- 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_digits = 150000, 20, 2, 1
+ seq, recorded_stack_counts = generate_sequences(
+ nb=nb,
+ nb_steps=nb_steps,
+ nb_stacks=nb_stacks,
+ nb_digits=nb_digits,
+ )
+
+ for n in range(min(10, seq.size(0))):
+ print(
+ seq_to_str(
+ seq[n],
+ nb_stacks=nb_stacks,
+ nb_digits=nb_digits,
+ recorded_stack_counts=recorded_stack_counts[n],
+ )
+ )
+ # print(seq_to_str(seq[n], nb_stacks=nb_stacks, nb_digits=nb_digits))
+
+ print("-- PREPARED FOR TEST -----------------")
+
+ remove_popped_values(seq, nb_stacks, nb_digits)
+
+ for n in range(min(10, seq.size(0))):
+ print(seq_to_str(seq[n], nb_stacks=nb_stacks, nb_digits=nb_digits))