/spiketraiNN

Lightweight Python library for spike train distance calculation

Primary LanguageC++MIT LicenseMIT

spiketraiNN: a lightweight Python library for spike train distance calculation.

Introduction

This is a Python wrapper for a set of spike train distance calculation routines in C++ (used implementation as well as generic DTW distance for time series, taken from here).

Available distance metrics include:

  • Victor-Purpura distance
  • Schreiber distance
  • ISI distance
  • SPIKE distance
  • Van Rossum distance
  • MaxMetric distance
  • ModulusMetric distance
  • DTW distance

Installation

make install

Usage

import numpy as np
import spiketrainn as spknn

n_points = 100
first_train = np.cumsum(np.random.uniform(10, 40, n_points))
second_train = np.cumsum(np.random.uniform(10, 40, n_points))

distance = spknn.distance(
    first_train, second_train, metric='isi'
)