Import supplements and repo reorganization

1 files changed, 16 insertions, 0 deletions

diff --git a/hawkes_experiments/sanity.py b/hawkes_experiments/sanity.py
new file mode 100644
index 0000000..b6f25eb
--- /dev/null
+++ b/hawkes_experiments/sanity.py
@@ -0,0 +1,16 @@
+from cPickle import load
+from csv import reader
+from data2 import parse
+import sys
+
+
+def parse_row(row):
+    return set(e for e in map(parse, row[2:]) if e)
+
+if __name__ == "__main__":
+    nodes, edges, events, event_edges = load(open("data2.pickle", "rb"))
+    with open(sys.argv[1]) as fh:
+        fh.readline()
+        reader = reader(fh)
+        d = {parse(row[1]): parse_row(row) for row in reader}
+    d = {k: v for (k, v) in d.iteritems() if v}