make processing easier by allowing users to specify dataset file

Author: prasys

dependency_graph.py

@@ -49,9 +49,14 @@ def process(filename):
     fout.close() 
 
 if __name__ == '__main__':
-    process('./datasets/acl-14-short-data/train.raw')
-    process('./datasets/acl-14-short-data/test.raw')
-    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')
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--dataset', default=None, type=str, help='path to dataset')
+    opt = parser.parse_args()
+    process(opt.dataset)
+
+    # 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/Laptops_Train.xml.seg')
+    # process('./datasets/semeval14/Laptops_Test_Gold.xml.seg')