core.transforms#

Unit selection and spike-count augmentations, shared by every task suite.

Unit transforms#

FilterUnits

Drops units that have failing probe QC or did not fire in this slice.

UnitDropout

A simple unit dropout transform.

apply_unit_mask_

Mask units and corresponding spikes given a boolean mask (in-place).

Augmentations#

AdditiveGaussianNoise

Adds Gaussian noise scaled to the input max, clipped at 0.

AdditivePepperNoise

Randomly zeros out bins.

AmplitudeScaling

Rescales amplitude by a small random factor.

GaussianNoise

Adds Gaussian noise scaled to the input std.

RandomApply

Apply a transform with probability p.

TemporalGaussianSmoothing

Gaussian smoothing along the temporal (last) axis only.

TemporalJittering

Random row-wise circular shift along the temporal axis.