Question about the dataset organization

[HowieMa]

Dear authors, Thanks for sharing such an amazing project. May I ask the organization of the training dataset and how to obtain the "train_keys.pkl" and "test_keys.pkl" in this code?

Looking forward to your reply. Thanks!