From 2bb0c04b2cc23d5bf850d5811657a83dd98767de Mon Sep 17 00:00:00 2001
From: opi <opitz@cl.uni-heidelberg.de>
Date: Tue, 23 Jul 2019 15:36:27 +0200
Subject: [PATCH] small bugfix and new argument : subcat - use subcategories
instead of categories
---
src/main.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/main.py b/src/main.py
index 5f9429e..05a841a 100644
--- a/src/main.py
+++ b/src/main.py
@@ -26,6 +26,9 @@ parser.add_argument('-verbose',choices=[0,1,2],type=int,default=0,
parser.add_argument('-prune_text_nodes_min_freq',type=int,default=2,
help='how often a text description node has to occur to be included in the graph')
+
+parser.add_argument('--subcategories',dest='subcat',action='store_true',
+ help='do we want subcategories (e.g. category punishment: corporal, subcategory: whipping), there will be many more nodes with subcat enabled')
args = parser.parse_args()
@@ -42,7 +45,7 @@ def write(path,G):
f.write(json.dumps(nx.readwrite.json_graph.node_link_data(G)))
-G,index_dict = gb.build_graph(years=args.year)
+G,index_dict = gb.build_graph(years=args.year,subcat=args.subcat)
print("Graph info before text description node simplification:")
print(nx.info(G))
gh.simplify_text_description_nodes(G,index_dict,mode=args.text_node_simplification_mode,min_freq=args.prune_text_nodes_min_freq)
--
GitLab