Sheshera Mysore commited on
Commit
8f8eff4
·
1 Parent(s): dcaab59

Exclude seed papers from results.

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -61,6 +61,9 @@ def read_user(seed_json):
61
  st.session_state['pid2sent_vectors_user'] = pid2sent_vectors_user
62
  st.session_state['user_kps'] = user_kps
63
  st.session_state['username'] = uname
 
 
 
64
  return doc_vectors_user, pid2idx_user, pid2sent_vectors, user_kps
65
  else:
66
  return st.session_state.doc_vectors_user, st.session_state.pid2idx_user, \
@@ -240,6 +243,10 @@ def second_stage_ranked_docs(selected_query_kps, first_stage_pids, pid2abstract,
240
  # Get the docs
241
  retrieved_papers = collections.OrderedDict()
242
  for pid in top_pids:
 
 
 
 
243
  retrieved_papers[pid2abstract[pid]['title']] = {
244
  'title': pid2abstract[pid]['title'],
245
  'kp_explanations': pid2kp_expls[pid],
 
61
  st.session_state['pid2sent_vectors_user'] = pid2sent_vectors_user
62
  st.session_state['user_kps'] = user_kps
63
  st.session_state['username'] = uname
64
+ for pd in seed_json['papers']:
65
+ norm_title = " ".join(pd['title'].lower().strip().split())
66
+ st.session_state.seed_titles.append(norm_title)
67
  return doc_vectors_user, pid2idx_user, pid2sent_vectors, user_kps
68
  else:
69
  return st.session_state.doc_vectors_user, st.session_state.pid2idx_user, \
 
243
  # Get the docs
244
  retrieved_papers = collections.OrderedDict()
245
  for pid in top_pids:
246
+ # Exclude papers from the seed set in the result set.
247
+ norm_title = " ".join(pid2abstract[pid]['title'].lower().strip().split())
248
+ if norm_title in st.session_state.seed_titles:
249
+ continue
250
  retrieved_papers[pid2abstract[pid]['title']] = {
251
  'title': pid2abstract[pid]['title'],
252
  'kp_explanations': pid2kp_expls[pid],