darpanaswal commited on
Commit
74714ba
·
verified ·
1 Parent(s): 5a33eb6

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +4 -4
cross_encoder_reranking_train.py CHANGED
@@ -145,10 +145,10 @@ def extract_text(content_dict, text_type="full"):
145
  filtered_dict = process_single_patent(content_dict)
146
  all_text = []
147
  # Start with abstract for better context at the beginning
148
- # if "title" in content_dict:
149
- # all_text.append(content_dict["title"])
150
- if "pa01" in content_dict:
151
- all_text.append(content_dict["pa01"])
152
 
153
  # For claims, paragraphs and features, we take only the top-10 most relevant
154
  # Add claims
 
145
  filtered_dict = process_single_patent(content_dict)
146
  all_text = []
147
  # Start with abstract for better context at the beginning
148
+ if "title" in content_dict:
149
+ all_text.append(content_dict["title"])
150
+ # if "pa01" in content_dict:
151
+ # all_text.append(content_dict["pa01"])
152
 
153
  # For claims, paragraphs and features, we take only the top-10 most relevant
154
  # Add claims