From bfae098bbc47b3948a2a794bee4e49ff32504d6b Mon Sep 17 00:00:00 2001 From: Thibaut Horel Date: Sat, 12 Sep 2015 16:37:44 -0400 Subject: Incorporate the time varying model for the background rate --- hawkes/data2.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'hawkes/data2.py') diff --git a/hawkes/data2.py b/hawkes/data2.py index c091e7a..b8f83c9 100644 --- a/hawkes/data2.py +++ b/hawkes/data2.py @@ -2,6 +2,7 @@ from csv import DictReader import sys from itertools import product from cPickle import dump +from math import cos MAX_TIME = 3012 @@ -10,13 +11,18 @@ def parse(s): return None if s == "NA" else int(float(s)) +def fluctuation_int(t): + if t is None: + t = MAX_TIME + return t + 0.43 / 0.0172 * (cos(4.36) - cos(0.0172 * t + 4.36)) + + def load_nodes(filename): with open(filename) as fh: reader = DictReader(fh) d = {parse(row["name"]): parse(row["fatal_day"]) for row in reader} for n, t in d.iteritems(): - if t is None: - d[n] = MAX_TIME + d[n] = fluctuation_int(t) return d -- cgit v1.2.3-70-g09d2