neuralbench.utils.make_weighted_sampler¶
- neuralbench.utils.make_weighted_sampler(dataset: SegmentDataset, logger: Logger) WeightedRandomSampler[source][source]¶
Create a weighted random sampler for the given dataset to handle class imbalance.
Back to top