I have my training data of shape (160, 17979). 160 is the size of the training data and 17979 is the feature number.
I reshaped my training data X to 3d using the fol