Test and Plot Spike Raster

class snnpytorch.test.test_spike_raster.Pipeline

Bases: object

Class for testing the network and plotting spike raster

gen_input_spike_raster() → None

Generate and save input spike raster.

load_dataset() → torch.Tensor

Load spike data for time points using torch dataloader

Returns

Input spike data for a single time point

parse_cmd() → None

Parse command line inputs and update the model

plot_simulation_results(input_spike_data: torch.Tensor, output_spike_data: torch.Tensor) → None

Plot input and output spike raster.

Parameters

• input_spike_data – Input spike raster

• output_spike_data – Output spike raster

run() -> (<class 'torch.Tensor'>, <class 'torch.Tensor'>)

Run the simulation for the defined number of steps.

Returns

input spike raster, output spike raster