cellflow.data.TrainSampler

class cellflow.data.TrainSampler(data, batch_size=1024)[source]

Data sampler for TrainingData.

Parameters:

  • data (TrainingData) – The training data.

  • batch_size (int) – The batch size.

Methods

sample(rng)

Sample a batch of data.

Attributes

data

The training data.