From 58faa01748fe0e6f6d040d1296266d17bd7a3543 Mon Sep 17 00:00:00 2001
From: Ben Green <bgreen@g.harvard.edu>
Date: Mon, 14 Sep 2015 23:19:34 -0400
Subject: prediction and plotting cascades

---
 R Scripts/predict-victims.R | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'R Scripts/predict-victims.R')

diff --git a/R Scripts/predict-victims.R b/R Scripts/predict-victims.R
index fc6cb7f..ccf894a 100644
--- a/R Scripts/predict-victims.R	
+++ b/R Scripts/predict-victims.R	
@@ -47,7 +47,7 @@ days = Reduce(union, list(lcc_verts$fatal_day,lcc_verts$nonfatal_day_1,
                               lcc_verts$nonfatal_day_4,lcc_verts$nonfatal_day_5))
 days = days[!is.na(days)]
 days = sort(days)
-days = split(days, ceiling(seq_along(days)/456))
+days = split(days, ceiling(seq_along(days)/92))
 lambdas = c(0, exp(seq(log(0.0000001), log(.95), length.out=100)), 1)
 
 ##### Loop through days
@@ -60,9 +60,7 @@ for(i in 1:length(days)){
   
     ##### Demographics model
     victims = lcc_verts
-    
     vics = which(victims$vic.day<day)
-    vic.days = victims$vic.day[vics]
     
     victims$age[-vics] = as.numeric(start_date + day - 1 - as.Date(victims$dob[-vics]))
     victims$arrests[-vics] = unlist(lapply(prior_arrests[-vics],nArrests,day=day))
@@ -86,7 +84,7 @@ for(i in 1:length(days)){
     irs = lcc_verts$ir_no[ids]
     
     risk = data.table(id=ids, ir=irs, weight=weights)
-    if (dim(risk)[1]>0) risk = risk[, list(weight=sum(weight)), by=ir] # max or sum
+    if (dim(risk)[1]>0) risk = risk[, list(weight=sum(weight)), by=ir]
   
     ##### Combined Model
     combined = dt
@@ -108,4 +106,4 @@ for(i in 1:length(days)){
   print(proc.time()-ptm)
 }
 
-# save(correct_rank, file='Results/correct_rank_62815.RData') 
\ No newline at end of file
+# save(correct_rank, file='Results/correct_rank_91415.RData') 
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2