preprocessing for tensorflow rnn with groups

Question

I am currently following the training https://www.tensorflow.org/tutorials/structured_data/time_series However, I am wondering how to modify the code to a dataset with multi