Update dependency_graph.py

Author: prasys

dependency_graph.py

@@ -3,6 +3,8 @@
 import numpy as np
 import spacy
 import pickle
+import argparse
+
 
 from spacy.tokens import Doc
 
@@ -56,7 +58,8 @@ def process(filename):
 
     # process('./datasets/acl-14-short-data/train.raw')
     # process('./datasets/acl-14-short-data/test.raw')
-    # process('./datasets/semeval14/Television_Train.xml.seg')
-    # process('./datasets/semeval14/Television_Test_Gold.xml.seg')
+    # process('./datasets/semeval14/Restaurants_Train.xml.seg')
+    # process('./datasets/semeval14/Restaurants_Test_Gold.xml.seg')
     # process('./datasets/semeval14/Laptops_Train.xml.seg')
-    # process('./datasets/semeval14/Laptops_Test_Gold.xml.seg')
+    # process('./datasets/semeval14/Laptops_Test_Gold.xml.seg')
+