Commit 653ffcbd authored by vdberg's avatar vdberg
Browse files

fix

parent 44a83560

process_srl.py

+2 −1
Original line number Diff line number Diff line
@@ -63,6 +63,7 @@ MIN_FREQ = args.min_freq 
N = args.n

ARG = args.arg

LEMMATIZE = args.lemmatize

print(args)



# load files

sent = pd.read_csv('data/sent_df.csv', names=['sent_id','bias','doc_id','s'])
@@ -76,7 +77,7 @@ output = output[output['name'] != 'Ryan'] 
#print('Arg-0 only {}:'.format(df.shape))



# combine output

print(output.sent_id.value_counts())

print(output.sent_id.value_counts().head)

output['sent_id'] = output['sent_id'].astype(float)

sent['sent_id'] = sent['sent_id'].astype(float)

df = pd.merge(output, sent, on=['sent_id'])