Commit 554f239e authored by Victor Zimmermann's avatar Victor Zimmermann
Browse files

Add dynamic minimum node/edge frequencies.

parent c1fba280
Loading
Loading
Loading
Loading
+8 −0
Original line number Diff line number Diff line
@@ -401,8 +401,16 @@ def build_graph(node_freq_dict: dict, edge_freq_dict: dict) -> nx.Graph:
            given target entity.
    """
    
    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()