hawkes_experiments/sanity.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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}