Sorry about this - also, when I try to convert the dset_train to a numpy array, it gives me the ‘Subset’ object has no attribute ‘numpy’’ error ?
I have also tried converting the trainset to numpy array before making subset, but that did not work either. It gives me a ‘ValueError: too many values to unpack (expected 2)’ error.