update float.

tests/test_model_spearman.py

@@ -337,6 +337,16 @@ def test_ernie3_0_base_model(self):
         # PAWSX spearman corr: 0.3428
         # avg: 0.6281
         # V100 QPS: 1526
+
+        # training with mean pooling and inference with mean pooling
+        # training data: all nli-zh-all sampled data
+        # STS-B spearman corr: 0.7742
+        # ATEC spearman corr: 0.4394
+        # BQ spearman corr: 0.6436
+        # LCQMC spearman corr: 0.7345
+        # PAWSX spearman corr: 0.3914
+        # avg: 0.5966
+        # V100 QPS: 1603
         pass
 
     def test_ernie3_0_xbase_model(self):

text2vec/cosent_dataset.py

@@ -35,7 +35,7 @@ def get_field_names(data_item):
             if not field1 or not field2:
                 continue
 
-            text_a, text_b, score = entry[field1], entry[field2], int(entry["label"])
+            text_a, text_b, score = entry[field1], entry[field2], float(entry["label"])
             data.append((text_a, score))
             data.append((text_b, score))
     else:
@@ -45,7 +45,7 @@ def get_field_names(data_item):
                 if len(line) != 3:
                     logger.warning(f'line size not match, pass: {line}')
                     continue
-                score = int(line[2])
+                score = float(line[2])
                 data.append((line[0], score))
                 data.append((line[1], score))
     return data