diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-01-13 23:49:05 -0600 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-01-13 23:49:05 -0600 |
| commit | cd99d6b874d9d09b3bb87b8485cc787885af71f1 (patch) | |
| tree | 59a233959932ca0e4f12f196275e07fcf443b33f /files/data_io/transforms/spike_augmentation.py | |
init commit
Diffstat (limited to 'files/data_io/transforms/spike_augmentation.py')
| -rw-r--r-- | files/data_io/transforms/spike_augmentation.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/files/data_io/transforms/spike_augmentation.py b/files/data_io/transforms/spike_augmentation.py new file mode 100644 index 0000000..9b7b687 --- /dev/null +++ b/files/data_io/transforms/spike_augmentation.py @@ -0,0 +1,10 @@ +import torch + +class SpikeJitter: + """Add temporal jitter noise to spikes.""" + def __init__(self, std=0.01): + self.std = std + + def __call__(self, spikes: torch.Tensor) -> torch.Tensor: + # TODO: add random jitter to spike timings + return spikes |