torch_brain.transforms

Compose	Compose several transforms together.
RandomChoice	Apply a single transformation randomly picked from a list.
ConditionalChoice	Conditionally apply a single transformation based on whether a condition is met.
UnitDropout	Augmentation that randomly drops units from the sample.
TriangleDistribution	Triangular distribution with a peak at mode_units, going from min_units to max_units.
RandomTimeScaling
RandomOutputSampler
RandomCrop
BinSpikes	Bin spike events into fixed-width time bins.
UnitFilter	Drop units based on the mask_fn given in the constructor.
UnitFilterById	Keep/drop units based on the keyword/regex given in the constructor.