summaryrefslogtreecommitdiffstats
path: root/experiments/build_network.py
diff options
context:
space:
mode:
authorBen Green <ben@SEASITs-MacBook-Pro.local>2015-06-11 21:27:14 -0400
committerBen Green <ben@SEASITs-MacBook-Pro.local>2015-06-11 21:27:14 -0400
commit1143cf60812c37957c81d39d6180921460e62901 (patch)
tree5231b7fe4f3e27d00072d652297208660e47bb5a /experiments/build_network.py
parent18aa2dd933f4c76ebd374b42fa10d0067870f135 (diff)
downloadcriminal_cascades-1143cf60812c37957c81d39d6180921460e62901.tar.gz
started adding weights into edge probabilities
Diffstat (limited to 'experiments/build_network.py')
-rw-r--r--experiments/build_network.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/experiments/build_network.py b/experiments/build_network.py
index 17c8bb5..47f657b 100644
--- a/experiments/build_network.py
+++ b/experiments/build_network.py
@@ -13,9 +13,7 @@ def build_network(filename):
for row in reader:
from_, to = int(float(row["from"])), int(float(row["to"]))
dist = int(row["dist"])
- w1 = int(row["w1"])
- w2 = int(row["w2"]) if dist>1 else float(row["w2"])
- w3 = int(row["w3"]) if dist>2 else float(row["w3"])
+ w1, w2, w3 = float(row["w1"]), float(row["w2"]), float(row["w3"])
if int(float(row["dist"])) > 2:
continue
# 'to' is a victim