darpanaswal commited on
Commit
c31c9ff
·
verified ·
1 Parent(s): 3d5863a

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +3 -3
cross_encoder_reranking_train.py CHANGED
@@ -195,12 +195,12 @@ def extract_text(content_dict, text_type="full"):
195
  # Add claims
196
  for claim in filtered_dict["claims"][:10]:
197
  all_text.append(claim)
198
- # Add paragraphs
199
- for paragraph in filtered_dict["paragraphs"][:10]:
200
- all_text.append(paragraph)
201
  # Add features
202
  for feature in filtered_dict["features"][:10]:
203
  all_text.append(feature)
 
 
 
204
 
205
  return " ".join(all_text)
206
 
 
195
  # Add claims
196
  for claim in filtered_dict["claims"][:10]:
197
  all_text.append(claim)
 
 
 
198
  # Add features
199
  for feature in filtered_dict["features"][:10]:
200
  all_text.append(feature)
201
+ # Add paragraphs
202
+ for paragraph in filtered_dict["paragraphs"][:10]:
203
+ all_text.append(paragraph)
204
 
205
  return " ".join(all_text)
206