Since you are using stratify=trainset.targets, you should get balanced splits as is described in the docs 1 of sklearn.model_selection.train_test_split.
If you want to manually create imbalanced splits, you could sample the desired amount of class indices from trainset.targets and create a new Subset using the new sample distribution.