Paula Leonova commited on
Commit
c7c8a10
1 Parent(s): c0937ef

Fix ground labels matching

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -268,12 +268,11 @@ if submit_button or example_button:
268
  lf_df['title'] = text_df['title'][i]
269
  labels_full_df = pd.concat([labels_full_df, lf_df[labels_full_col_list]])
270
 
271
- with st.expander(f'({i+1}/{len(text_df)}) See intermediate label matching results'):
272
- st.write(f"Results for: {text_df['title'][i]}")
273
  if gen_summary == 'Yes':
274
- st.dataframe(pd.merge(labels_sum_df, labels_full_df, on=['title','label']))
275
  else:
276
- st.dataframe(labels_full_df)
277
 
278
  if gen_summary == 'Yes':
279
  label_match_df = pd.merge(labels_sum_df, labels_full_df, on=title_element + ['label'])
@@ -286,13 +285,15 @@ if submit_button or example_button:
286
  elif uploaded_onetext_glabels_file is not None:
287
  gdata = pd.read_csv(uploaded_onetext_glabels_file, header=None)
288
  join_list = ['label']
 
289
  elif uploaded_multitext_glabels_file is not None:
290
  gdata = pd.read_csv(uploaded_multitext_glabels_file)
291
  join_list = ['title', 'label']
 
292
 
293
  if len(glabels) > 0 or uploaded_onetext_glabels_file is not None or uploaded_multitext_glabels_file is not None:
294
  gdata['correct_match'] = True
295
- label_match_df = pd.merge(label_match_df, gdata, how='outer', on=join_list)
296
  label_match_df['correct_match'].fillna(False, inplace=True)
297
 
298
  st.dataframe(label_match_df)
 
268
  lf_df['title'] = text_df['title'][i]
269
  labels_full_df = pd.concat([labels_full_df, lf_df[labels_full_col_list]])
270
 
271
+ with st.expander(f'({i+1}/{len(text_df)}) See intermediate label matching results for: {text_df["title"][i]}'):
 
272
  if gen_summary == 'Yes':
273
+ st.dataframe(pd.merge(ls_df, lf_df, on=['label']))
274
  else:
275
+ st.dataframe(lf_df)
276
 
277
  if gen_summary == 'Yes':
278
  label_match_df = pd.merge(labels_sum_df, labels_full_df, on=title_element + ['label'])
 
285
  elif uploaded_onetext_glabels_file is not None:
286
  gdata = pd.read_csv(uploaded_onetext_glabels_file, header=None)
287
  join_list = ['label']
288
+ gdata.columns = join_list
289
  elif uploaded_multitext_glabels_file is not None:
290
  gdata = pd.read_csv(uploaded_multitext_glabels_file)
291
  join_list = ['title', 'label']
292
+ gdata.columns = join_list
293
 
294
  if len(glabels) > 0 or uploaded_onetext_glabels_file is not None or uploaded_multitext_glabels_file is not None:
295
  gdata['correct_match'] = True
296
+ label_match_df = pd.merge(label_match_df, gdata, how='left', on=join_list)
297
  label_match_df['correct_match'].fillna(False, inplace=True)
298
 
299
  st.dataframe(label_match_df)