diff options
Diffstat (limited to 'src/cascade_creation.py')
| -rw-r--r-- | src/cascade_creation.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/cascade_creation.py b/src/cascade_creation.py index 39fc531..db0ea7a 100644 --- a/src/cascade_creation.py +++ b/src/cascade_creation.py @@ -9,8 +9,9 @@ class InfluenceGraph(nx.DiGraph): """ networkX graph with mat and logmat attributes """ - def __init__(self, max_proba, *args, **kwargs): + def __init__(self, max_proba=None, min_proba=None, *args, **kwargs): self.max_proba = max_proba + self.min_proba = min_proba super(InfluenceGraph, self).__init__(*args, **kwargs) def erdos_init(self, n, p): @@ -142,8 +143,10 @@ def add_edges_from_proba_vector(G, p_node, node, floor_cstt): Also updates G's mat matrix """ floor_parent = np.nonzero(p_node*(p_node > floor_cstt)) + print(floor_parent) for parent in floor_parent[0]: - G.add_edge(parent, node) + #SOMEHOW THERE WAS A BUG HERE! + G.add_edge(node, parent) #TODO: update G's mat matrix return G |