Struct SubsetRandomSampler 

pub struct SubsetRandomSampler { /* private fields */ }

Random sampler that samples from a subset of indices

This sampler takes a predefined subset of indices and samples from them in random order. Useful for creating custom data splits or working with filtered datasets.

Examples

use torsh_data::sampler::{SubsetRandomSampler, Sampler};

// Sample from odd indices only
let subset_indices = vec![1, 3, 5, 7, 9];
let sampler = SubsetRandomSampler::new(subset_indices).with_generator(42);

let indices: Vec<usize> = sampler.iter().collect();
assert_eq!(indices.len(), 5);

Implementations

impl SubsetRandomSampler

pub fn new(indices: Vec<usize>) -> Self

Create a new subset random sampler

Arguments

• indices - The subset of indices to sample from

pub fn with_generator(self, seed: u64) -> Self

Set random generator seed

Arguments

• seed - Random seed for reproducible sampling

pub fn indices(&self) -> &[usize]

Get the subset indices

pub fn generator(&self) -> Option<u64>

Get the generator seed if set

Trait Implementations

impl Clone for SubsetRandomSampler

fn clone(&self) -> SubsetRandomSampler

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for SubsetRandomSampler

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Sampler for SubsetRandomSampler

type Iter = SamplerIterator

Iterator type returned by the sampler

fn iter(&self) -> Self::Iter

Create an iterator over indices

fn len(&self) -> usize

Total number of samples that will be yielded

fn is_empty(&self) -> bool

Check if sampler is empty

fn into_batch_sampler( self, batch_size: usize, drop_last: bool, ) -> BatchingSampler<Self>

where Self: Sized,

Convert this sampler into a batch sampler

fn into_distributed( self, num_replicas: usize, rank: usize, ) -> DistributedWrapper<Self>

where Self: Sized,

Create a distributed version of this sampler