From 5e6a49538d771fed753806a2efafbf878ce9e30e Mon Sep 17 00:00:00 2001
From: zimmermann <zimmermann@cl.uni-heidelberg.de>
Date: Thu, 22 Mar 2018 18:48:41 +0100
Subject: [PATCH] Formatting.
---
src/absinth.py | 11 +++++++++++
src/abstinent.py | 11 +++++++++++
2 files changed, 22 insertions(+)
diff --git a/src/absinth.py b/src/absinth.py
index 02de6f2..fb40141 100644
--- a/src/absinth.py
+++ b/src/absinth.py
@@ -148,6 +148,8 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict):
node_freq_dict = json.load(node_file)
edge_freq_dict = json.load(edge_file)
+ edge_freq_dict = {(key1,key2):value2 for key1,value1 in edge_freq_dict.items() for key2,value2 in value1.items()}
+
continue
if graph_in_existence == False:
@@ -157,6 +159,15 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict):
with open(config.graph+node_dict_name, 'w') as node_file, open(config.graph+edge_dict_name, 'w') as edge_file:
json.dump(node_freq_dict, node_file)
+
+ new_dict = dict()
+ for key,value in edge_freq_dict.items():
+ if key[0] in new_dict:
+ key[0].add({key[1]:value})
+ else:
+ key[0] = {key[1]:value}
+
+ edge_freq_dict = new_dict
json.dump(edge_freq_dict, edge_file)
#builds graph from these dictionaries, also applies multiple filters
diff --git a/src/abstinent.py b/src/abstinent.py
index 29201f1..901b99e 100644
--- a/src/abstinent.py
+++ b/src/abstinent.py
@@ -366,6 +366,8 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict):
node_freq_dict = json.load(node_file)
edge_freq_dict = json.load(edge_file)
+ edge_freq_dict = {(key1,key2):value2 for key1,value1 in edge_freq_dict.items() for key2,value2 in value1.items()}
+
continue
if graph_in_existence == False:
@@ -375,6 +377,15 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict):
with open(config.graph+node_dict_name, 'w') as node_file, open(config.graph+edge_dict_name, 'w') as edge_file:
json.dump(node_freq_dict, node_file)
+
+ new_dict = dict()
+ for key,value in edge_freq_dict.items():
+ if key[0] in new_dict:
+ key[0].add({key[1]:value})
+ else:
+ key[0] = {key[1]:value}
+
+ edge_freq_dict = new_dict
json.dump(edge_freq_dict, edge_file)
#builds graph from these dictionaries, also applies multiple filters
--
GitLab