Hi AgnesHB,
Maybe this will help? I used Sequence() to count from 1 to 5 (with step size of 1, and repeating each value 1 time), and placed the sequence in a match function to recode 1, 2, and 3 as 0 (for training), 4 as 1 (for validation) and 5 as 2 (for test), which will fit your .6, .2, .2 setup.
JSL: Match(Sequence(1, 5, 1, 1), 1, 0, 2, 0, 3, 0, 4, 1, 5, 2)

I hope this helps!
Julian