Commit 29ed661c authored by vdberg's avatar vdberg
Browse files

added range option

parent 6faa4ffb

process_srl.py

+0 −2
Original line number Diff line number Diff line
@@ -132,8 +132,6 @@ for n, gr in df.groupby(['bias', 'name_bias']): 
    sorted_verbs = counts.sort_values(by="normalized", ascending=False).head(N)

    sorted_verbs.to_csv("frequent_verbs/{}{}.csv".format(n_edit, '_arg' if ARG else ''), sep="\t")

    binned[n] = sorted_verbs.index

    binned[n + 'across'] = sorted_verbs.across

    binned[n + 'across'] = sorted_verbs.across



print(binned.head(N))

binned.to_csv('frequent_verbs/most_freq_verbs{}.csv'.format('_arg' if ARG else ''))