Spaces:
Configuration error
Configuration error
Update cross_encoder_reranking_train.py
Browse files
cross_encoder_reranking_train.py
CHANGED
|
@@ -134,10 +134,8 @@ def extract_text(content_dict, text_type="full"):
|
|
| 134 |
for key, value in content_dict.items():
|
| 135 |
if key.startswith('c-'):
|
| 136 |
content.append(value)
|
| 137 |
-
|
| 138 |
-
|
| 139 |
-
content.append(value)
|
| 140 |
-
return " ".join(content)
|
| 141 |
|
| 142 |
elif text_type == "tac1":
|
| 143 |
# Extract title, abstract, and first claim
|
|
|
|
| 134 |
for key, value in content_dict.items():
|
| 135 |
if key.startswith('c-'):
|
| 136 |
content.append(value)
|
| 137 |
+
if key == "features":
|
| 138 |
+
content += list(content_dict[key].values())
|
|
|
|
|
|
|
| 139 |
|
| 140 |
elif text_type == "tac1":
|
| 141 |
# Extract title, abstract, and first claim
|