Update dataloader.py

update sent2doc by adding tffrac. See #19

Author: dqwang122

module/dataloader.py

@@ -382,7 +382,7 @@ def CreateGraph(self, docLen, sent_pad, doc_pad, label, w2s_w, w2d_w):
             # s2d
             docid = sent2doc[i]
             docnid = docid2nid[docid]
-            G.add_edge(sent_nid, docnid, data={"dtype": torch.Tensor([2])})
+            G.add_edge(sent_nid, docnid, data={"tffrac": torch.LongTensor([0]), "dtype": torch.Tensor([2])})
 
         # add doc edges
         for i in range(article_num):