From 75868035a6e363c2ad67ff0342440aff3972a970 Mon Sep 17 00:00:00 2001
From: Ben Green <bgreen@g.harvard.edu>
Date: Sun, 4 Oct 2015 17:18:05 -0400
Subject: ..

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

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

diff --git a/R Scripts/predict-victims.R b/R Scripts/predict-victims.R
index d17ef83..3cffcbe 100644
--- a/R Scripts/predict-victims.R	
+++ b/R Scripts/predict-victims.R	
@@ -22,7 +22,8 @@ edges_all = data.table(dag_dat_lcc)
 vic_ids = which(lcc_verts$vic==T)
 lcc_verts = lcc_verts[,c('id','ir_no','nonfatal_day_1','nonfatal_day_2',
                          'nonfatal_day_3','nonfatal_day_4','nonfatal_day_5',
-                         'fatal_day','sex','race','dob','gang.member','gang.name')]
+                         'fatal_day','sex','race','dob','gang.member',
+                         'gang.name','district')]
 lcc_verts$vic.day = NA
 lcc_verts$vic.day[vic_ids] = as.numeric(apply(vic_times_lcc[vic_ids,2:7],1,min,na.rm=T))
 lcc_verts$age = FALSE
@@ -35,10 +36,11 @@ vic_times_lcc$ir_no = lcc_verts$ir_no
 rownames(vic_times_lcc) = vic_times_lcc$ir_no
 
 ##### Initialize data
-formula = vic ~ sex + race + age + gang.member + gang.name + arrests
+formula = vic ~ sex + race + age + gang.member + gang.name + arrests + district
 lcc_verts$sex = as.factor(lcc_verts$sex)
 lcc_verts$race = as.factor(lcc_verts$race)
 lcc_verts$gang.name = as.factor(lcc_verts$gang.name)
+lcc_verts$district = as.factor(lcc_verts$district)
 dt = data.table(ir=lcc_verts$ir_no, dem=0, cas=0, comb=0)
 
 #####
@@ -48,7 +50,7 @@ days = Reduce(union, list(lcc_verts$fatal_day,lcc_verts$nonfatal_day_1,
 days = days[!is.na(days)]
 days = sort(days)
 days = split(days, ceiling(seq_along(days)/92))
-lambdas = c(0, exp(seq(log(0.0001), log(0.01), length.out=100)), 1)
+lambdas = c(0, exp(seq(log(0.00001), log(0.01), length.out=100)), 1)
 
 ##### Loop through days
 correct_rank = c()
-- 
cgit v1.2.3-70-g09d2