summaryrefslogtreecommitdiffstats
path: root/experiments/build_network.py
diff options
context:
space:
mode:
authorBen Green <ben@SEASITs-MacBook-Pro.local>2015-06-15 17:15:33 -0400
committerBen Green <ben@SEASITs-MacBook-Pro.local>2015-06-15 17:15:33 -0400
commitf7997e0c67ea85b42da94894b6a6f9bef7382170 (patch)
tree57ac35463e4d45a84c0e1f215793519023a9ccbc /experiments/build_network.py
parent1143cf60812c37957c81d39d6180921460e62901 (diff)
downloadcriminal_cascades-f7997e0c67ea85b42da94894b6a6f9bef7382170.tar.gz
minor changes for testing
Diffstat (limited to 'experiments/build_network.py')
-rw-r--r--experiments/build_network.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/experiments/build_network.py b/experiments/build_network.py
index 47f657b..fd7179f 100644
--- a/experiments/build_network.py
+++ b/experiments/build_network.py
@@ -14,8 +14,8 @@ def build_network(filename):
from_, to = int(float(row["from"])), int(float(row["to"]))
dist = int(row["dist"])
w1, w2, w3 = float(row["w1"]), float(row["w2"]), float(row["w3"])
- if int(float(row["dist"])) > 2:
- continue
+ # if int(float(row["dist"])) > 1:
+ # continue
# 'to' is a victim
if row["t2"] != "NA":
dt = int(row["t2"]) - int(row["t1"])
@@ -32,7 +32,7 @@ def build_network(filename):
dt = 3012 - int(row["t1"])
parent = (dist, dt, w1, w2, w3)
if to not in non_victims:
- age += 3012 - int(row["spawn2"])
+ # age += 3012 - int(row["spawn2"])
non_victims[to] = []
non_victims[to].append(parent)
if from_ not in victims:
@@ -53,4 +53,5 @@ if __name__ == "__main__":
filename = sys.argv[1]
root, _ = splitext(filename)
root_victims, victims, non_victims, age = build_network(filename)
+ print age
dump((root_victims, victims, non_victims, age), open(root + ".pickle", "w"))