Redo Fig. 1b (#9)

* Redo Fig. 1b * updates * update * Update utilities.py * improve readability
nwlandry · Nov 6, 2023 · c056da3 · c056da3
1 parent ef8e947
commit c056da3
Show file tree

Hide file tree

Showing 5 changed files with 111 additions and 63 deletions.
diff --git a/Figures/contagion_inference.pdf b/Figures/contagion_inference.pdf
diff --git a/Figures/contagion_inference.png b/Figures/contagion_inference.png
diff --git a/lcs/utilities.py b/lcs/utilities.py
@@ -1,6 +1,7 @@
 import random
 
 import numpy as np
+from scipy.stats import gaussian_kde
 
 from .contagion import *
 from .inference import *
@@ -66,9 +67,9 @@ def degrees(A):
 
 def powerlaw(n, minval, maxval, r):
     u = np.random.random(n)
-    return (minval ** (1 - r) + u * (maxval ** (1 - r) - minval ** (1 - r))) ** (
-        1 / (1 - r)
-    )
+    a = minval ** (1 - r)
+    b = maxval ** (1 - r)
+    return (a + u * (b - a)) ** (1 / (1 - r))
 
 
 def mean_power_law(minval, maxval, r):

diff --git a/plot_infer_contagion_functions.ipynb b/plot_infer_contagion_functions.ipynb
diff --git a/requirements.txt b/requirements.txt
@@ -4,4 +4,5 @@ networkx>=3.0
 matplotlib
 numba
 pytest
-xgi
+xgi
+arviz
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,4 +4,5 @@ networkx>=3.0 @@
     matplotlib
     numba
     pytest
-    xgi
+    xgi
+    arviz