Bases: ConnectomeDataset
Reader for Varshney et al. 2011 connectivity dataset
Source code in cect/VarshneyDataReader.py
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78 | class VarshneyDataReader(ConnectomeDataset):
"""Reader for Varshney et al. 2011 connectivity dataset"""
def __init__(self):
ConnectomeDataset.__init__(self)
cells, neuron_conns = self.read_data()
for conn in neuron_conns:
self.add_connection_info(conn)
def read_data(self):
cells = []
conns = []
wb = load_workbook(filename)
sheet = wb.worksheets[0]
print_("Opened the Excel file: " + filename)
for row in sheet.iter_rows(
min_row=2, values_only=True
): # Assuming data starts from the second row
pre = str(row[0])
post = str(row[1])
if not post == NMJ_ENDPOINT:
syntype = str(row[2])
num = int(row[3])
synclass = (
GENERIC_ELEC_SYN
if syntype == "EJ"
else GENERIC_CHEM_SYN
if (syntype == "Sp" or syntype == "S")
else None
)
if synclass is not None:
conns.append(ConnectionInfo(pre, post, num, syntype, synclass))
if pre not in cells:
cells.append(pre)
if post not in cells:
cells.append(post)
return cells, conns
def read_muscle_data(self):
conns = []
neurons = []
muscles = []
return neurons, muscles, conns
|