rproepp / spykeutils Goto Github PK

import quantities as pq
import spykeutils.spike_train_generation as stg
import spykeutils.spike_train_metrics as stm
import pymuvr

n_observations = 4
n_cells = 20
rate = 30
tstop = 2
cos = 0.02
tau = 0
sutils_units = {}
pymuvr_observations = []

for unit in range(n_cells):
    sutils_units[unit] = []
    for ob in range(n_observations):
        sutils_units[unit].append(stg.gen_homogeneous_poisson(rate * pq.Hz, t_stop=tstop * pq.s))
    # observation 1 is identical to observation 0 for all the cells.
for unit in range(n_cells):
    sutils_units[unit][1] = sutils_units[unit][0]
for ob in range(n_observations):
    pymuvr_observations.append([])
    for unit in range(n_cells):
        pymuvr_observations[ob].append(sutils_units[unit][ob].tolist())

sutils_d = stm.van_rossum_multiunit_dist(sutils_units,
                                         weighting=cos,
                                         tau=tau)
pymuvr_d = pymuvr.square_distance_matrix(pymuvr_observations,
                                         cos,
                                         tau)

print("\nspykeutils result:")
print(sutils_d)

print"\npymuvr result:"
print(pymuvr_d)