Commit 5f0668b7 authored by vdberg's avatar vdberg
Browse files

still problems

parent deffeda4

srl.py

+5 −3
Original line number Diff line number Diff line
@@ -80,18 +80,20 @@ for i, r in data.iterrows(): 
        #print('verbs', verbs)

        #names['nambias'] = namlabs_df.loc[verbs.index].bias.values

        verbs['same'] = namlabs_df.loc[verbs.index].bias.values == r.bias

        print('verbs', verbs)

        #print('verbs', verbs)

        verbs['id'] = [i] * len(verbs)

        #verbs.arg = verbs.arg.str[2:]

        verb_df = verb_df.append(verbs[['arg', 'verb', 'same', 'id']])

        verb_df = verb_df.append(verbs)

        print('Found {} verbs'.format(len(verbs)))

        verb_df.to_csv('vdberg_output.csv')

        verbs.to_csv('outputs/vdberg_output_{}.csv'.format(i))



    t1 = time.time()

    total_n = t1-t0

    times.append(total_n)

    print('Took {} seconds'.format(sum(times)/len(times)))



verb_df.to_csv('vdberg_output.csv')





#grs = verb_df.groupby('same')

#for n, gr in grs: