Commit 1fb9ddc2 authored by vdberg's avatar vdberg
Browse files

bug

parent 3f1419c3

process_srl.py

+2 −2
Original line number Diff line number Diff line
@@ -87,8 +87,8 @@ def cust(x): 
grs = pd.DataFrame()

for n, gr in df.groupby(['bias', 'name_bias']):

    counts = gr.verb.value_counts()

    counts = counts / overall_counts.loc[counts.index]

    counts.index += counts.apply(cust)

    counts2 = counts / overall_counts.loc[counts.index]

    counts.index += counts2.apply(cust)

    sorted_verbs = counts.sort_values(ascending=False).head(args.n)

    grs[n] = sorted_verbs.index

print(grs)