Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -128,6 +128,18 @@ def extract_text(content_dict, text_type="full"):
                 claims.append(value)
         return " ".join(claims)
 
+    elif text_type == "claimfeat":
+        # Extract all claims (keys starting with 'c')
+        content = []
+        for key, value in content_dict.items():
+            if key.startswith('c-'):
+                claims.append(value)
+
+        for key, value in content_dict.items():
+            for key1, value1 in content_dict["features"]:
+                claims.append(value1)
+        return " ".join(content)
+
     elif text_type == "tac1":
         # Extract title, abstract, and first claim
         title = content_dict.get("title", "")