Commit f17398e3 authored by EstherMaria's avatar EstherMaria
Browse files
parents 1eb0a47c 7058ac0d
Loading
Loading
Loading
Loading
+4 −1
Original line number Diff line number Diff line
@@ -44,12 +44,15 @@ parser.add_argument("-c", "--combine", action="store_true", default=False, help=
args = parser.parse_args()
df = load_output(dir='outputs', combine=args.combine)

df = df[df['arg'].str.startswith('B-ARG0')]

print('Output size {}:'.format((df.shape))

sent = pd.read_csv('data/sent_df.csv', names=['sent_id','bias','doc_id','s'])
df = pd.merge(df, sent, on=['sent_id'])
namlabs_df, namlabs = load_nameslab()
df = pd.merge(df, namlabs_df, on=['name'])
#df = df.drop(columns='same')
df = df[df['arg'].str.startswith('B-ARG0')]

pd.set_option('display.max_colwidth', 200)
pd.set_option('display.max_columns', 10)