fix parameter

xunzheng · Nov 11, 2018 · 6703c72 · 6703c72
1 parent b1d6760
commit 6703c72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -28,7 +28,7 @@ def simulate_random_dag(d: int,
  B = np.tril((np.random.rand(d, d) < prob).astype(float), k=-1)
  elif graph_type == 'barabasi-albert':
  m = int(round(degree / 2))
- B = np.zeros(d, d)
+ B = np.zeros([d, d])
  bag = [0]
  for ii in range(1, d):
  dest = np.random.choice(bag, size=m)