Skip to main content

module spikedetector.py

Global Variables​

  • model_path
  • map_location

class SpikeDetector​

function __init__​

__init__(model_path, n_channels=256, n_frames=200, device='cuda')

function detect​

detect(data_frame)

Takes in the current frame of data, adds it to the data slice, and returns the spike predictions

Parameters ---------- data_frame : np.array of shape (n_channels, 1) The current frame of data Returns ------- spike_preds : np.array of shape (n_channels, 120) The spike predictions for each channel, for the previous 6ms, or 120 frames

function detect_chunk​

detect_chunk(data_chunk)

Takes in a chunk of data, and returns the spike predictions for each channel for the last 6ms of data

Parameters ---------- data_chunk : np.array of shape (n_channels, n_frames) The chunk of data to be processed Returns ------- spike_preds : np.array of shape (n_channels, 120) The spike predictions for each channel, for the previous 6ms, or 120 frames

function reset​

reset()

This file was automatically generated via lazydocs.