Spaces:
Runtime error
Runtime error
Sheshera Mysore
commited on
Commit
·
8f8eff4
1
Parent(s):
dcaab59
Exclude seed papers from results.
Browse files
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],
|