diff options
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")) |