Stratified Multilabel Dataset in Pytorch

Is there any method in Pytorch like train_test_split in sklearn to stratify the dataset into train, valid and test for multilabel classifications?

No, I don’t think there is a built-in PyTorch method, but you could of course use the scikit-learn methods and I would also recommend to use already implemented and tested methods from the Python ecosystem, if possible.

Okay, thanks for your reply!