poster+beginning of experiments

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