darpanaswal commited on
Commit
fb46599
·
verified ·
1 Parent(s): 80473ca

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +2 -2
cross_encoder_reranking_train.py CHANGED
@@ -145,8 +145,8 @@ 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 "pa01" in content_dict:
149
- # all_text.append(content_dict["pa01"])
150
 
151
  # For claims, paragraphs and features, we take only the top-10 most relevant
152
  # 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 "pa01" in content_dict:
149
+ all_text.append(content_dict["pa01"])
150
 
151
  # For claims, paragraphs and features, we take only the top-10 most relevant
152
  # Add claims