How to do RUS and ROS on imbalanced datasets

vision
2

You could use the WeightedRandomSampler to over- or undersample the dataset.
Here is an example showing how to use it. :slight_smile: