From ab0b1f3cefedb35327a19ec1b6afd560bfdf802d Mon Sep 17 00:00:00 2001
From: Thibaut Horel <thibaut.horel@gmail.com>
Date: Mon, 14 Sep 2015 23:08:02 -0400
Subject: Import supplements and repo reorganization

---
 ic_experiments/new.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 ic_experiments/new.py

(limited to 'ic_experiments/new.py')

diff --git a/ic_experiments/new.py b/ic_experiments/new.py
new file mode 100644
index 0000000..4274dd7
--- /dev/null
+++ b/ic_experiments/new.py
@@ -0,0 +1,23 @@
+from math import log, exp
+
+T = 100
+N = 100
+
+
+def kernel(t, mu):
+    return mu * exp(-mu * t)
+
+
+def base_rate(t, lamb):
+    return lamb
+
+
+def ll(crimes, weights, mu, lamb):
+    r = 0
+    for i, crime in enumerate(crimes):
+        t, v = crime
+        a = sum(weights[(u, v)] * kernel(t - s, mu) for s, u in crimes[:t])
+        r += log(base_rate(t, lamb) + a)
+    for j in range(N):
+        a = sum(weights[(u, v)] * kernel(T - s, mu) for s, u in crimes)
+        r -= log(a)
-- 
cgit v1.2.3-70-g09d2