This is the code that I need to configure, actually I have directly splitted data so I do not need to split my dataset I can remove that parts, I have train_features.npy, tr
