use e-s-c as positive examples

finetune.py

@@ -25,12 +25,21 @@ class ESCIDataset(Dataset):
         with gzip.open(input) as jsonfile:
             for line in jsonfile.readlines():
                 query = json.loads(line)
-                if len(query['e']) > 0 and len(query['i']) > 0:
-                    for p in query['e']:
-                        positive = p['title']
-                        for n in query['i']:
-                            negative = n['title']
-                            self.queries.append(InputExample(texts=[query['query'], positive, negative]))
+                for p in query['e']:
+                    positive = p['title']
+                    for n in query['i']:
+                        negative = n['title']
+                        self.queries.append(InputExample(texts=[query['query'], positive, negative]))
+                for p in query['s']:
+                    positive = p['title']
+                    for n in query['i']:
+                        negative = n['title']
+                        self.queries.append(InputExample(texts=[query['query'], positive, negative]))
+                for p in query['c']:
+                    positive = p['title']
+                    for n in query['i']:
+                        negative = n['title']
+                        self.queries.append(InputExample(texts=[query['query'], positive, negative]))
 
     def __getitem__(self, item):
         return self.queries[item]

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:089d832e391c446f9bc7df2f9b5f772405a7e762506908f9a22b6a71fb172e74
+oid sha256:32be7d4ded580fffe92bfff7e8dc865b17dc0cfbba3dd598865adc54dd89d0c3
 size 90891565

pytorch_model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cfa7e76b1b76b1617d7188ce710c8f03ac0c85463f8ad55b276c484d2c7c725a
+oid sha256:b1abb2047d142c643ed10c33b6a1517171dab1ffb3d3a57bc2043437d3f5bf77
 size 90984263