Skip to content
Snippets Groups Projects
Commit 554f239e authored by Victor Zimmermann's avatar Victor Zimmermann
Browse files

Add dynamic minimum node/edge frequencies.

parent c1fba280
No related branches found
No related tags found
No related merge requests found
......@@ -401,8 +401,16 @@ def build_graph(node_freq_dict: dict, edge_freq_dict: dict) -> nx.Graph:
given target entity.
"""
min_node_freq = config.min_node_freq
min_edge_freq = config.min_edge_freq
if config.use_mean_freq:
min_node_freq = np.mean(node_freq_dict.values())
min_edge_freq = np.mean(edge_freq_dict.values())
else:
min_node_freq = config.min_node_freq
min_edge_freq = config.min_edge_freq
max_weight = config.max_weight
cooccurrence_graph = nx.Graph()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment