diff options
| author | Thibaut Horel <thibaut.horel@gmail.com> | 2015-09-14 22:45:10 -0400 |
|---|---|---|
| committer | Thibaut Horel <thibaut.horel@gmail.com> | 2015-09-14 22:45:10 -0400 |
| commit | 6fdfdbbc59f758459b9bcb992a950c99722e50c1 (patch) | |
| tree | 6332b50463da5ac70038d473aae28d8d170b6042 /hawkes/data.py | |
| parent | 0cd7df0103230be0b73f888a6b5553440ef40d03 (diff) | |
| download | criminal_cascades-6fdfdbbc59f758459b9bcb992a950c99722e50c1.tar.gz | |
Update refine accordingly
Diffstat (limited to 'hawkes/data.py')
| -rw-r--r-- | hawkes/data.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/hawkes/data.py b/hawkes/data.py index 5f68dd1..0f6135b 100644 --- a/hawkes/data.py +++ b/hawkes/data.py @@ -34,8 +34,6 @@ def load_edges(filename): for row in reader: fro, to, t, dist = map(parse, [row["from"], row["to"], row["t1"], row["dist"]]) - if dist >= 2: - continue d = edges.get(fro, dict()) d[to] = dist edges[fro] = d @@ -70,4 +68,4 @@ if __name__ == "__main__": nodes = load_nodes(sys.argv[1]) edges, events = load_edges(sys.argv[2]) event_edges = compute_event_edges(events, edges) - dump((nodes, edges, events, event_edges), open("data-dist1.pickle", "wb")) + dump((nodes, edges, events, event_edges), open("data-all.pickle", "wb")) |