Get label as the last entry in the tensor dataset

[rdisipio]

imbalanced-dataset-sampler/torchsampler/imbalanced.py

Line 51 in 01cb129

return dataset.tensors[1]

I think this should be return dataset.tensors[-1] . I have run into issues when the TensorDataset object is composed of more than 1 dataframe, with the last one corresponding to labels.