diff options
Diffstat (limited to 'src/cascade_creation.py')
| -rw-r--r-- | src/cascade_creation.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/cascade_creation.py b/src/cascade_creation.py index cdf7484..714c120 100644 --- a/src/cascade_creation.py +++ b/src/cascade_creation.py @@ -65,7 +65,8 @@ class InfluenceGraph(nx.DiGraph): if not hasattr(self, '_mat'): if self.min_proba is None or self.max_proba is None: print("You forgot to initialize the min and max proba") - self._mat = np.zeros((self.number_of_nodes(), self.number_of_nodes())) + self._mat = np.zeros((self.number_of_nodes(), + self.number_of_nodes())) else: self._mat = self.max_proba * np.random.rand(len(self), len(self) ) * np.asarray(nx.adjacency_matrix(self).todense().T) @@ -73,7 +74,8 @@ class InfluenceGraph(nx.DiGraph): print("adding sparse edges to the graph") #Adding sparse non-edges to the graph! self._mat += .1 * np.random.rand(len(self), len(self) - ) * np.random.binomial(1, p=.33, size=(len(self), len(self))) + ) * np.random.binomial(1, p=.33, size=(len(self), + len(self))) return self._mat @property |