1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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 | #1 -- get dialogues
finals = []
dialogues = cands.dialogue_num.drop_duplicates()
for d in tqdm(dialogues):
# 2 -- get segment list
seg_list = get_seg_list(cands[cands.dialogue_num == d])
# 3 -- create seg pairs list
seg_pairs = []
for i, s in enumerate(seg_list):
for n in [j for j in reversed(range((i+1)-20, i+1)) if j>=0]:
try:
seg_pairs.append((seg_list[n], seg_list[i+1]))
except IndexError:
pass
# 4 -- for each pair, pull row and append rules
for s in seg_pairs:
row = cands[(cands.source_id == s[0]) & (cands.target_id == s[1])]
if row.empty:
continue
else:
r_index = row.index[0]
row = row.iloc[0]
# make sure that if any backwards links exist they are also 0
backwards = cands[(cands.source_id == s[1]) & (cands.target_id == s[0])]
if backwards.shape[0] > 0:
b_index = backwards.index[0]
finals.append((b_index, -1))
#we only look at ling --> ling cases
if row.source_type != 'Segment' or row.target_type != 'Segment':
link = 0
else:
link_one = LF_Contrast_L_L(row)
if link_one:
link = 1
else:
link = 0
finals.append((r_index, link))
|