diff options
Diffstat (limited to 'R Scripts/predict-victims.R')
| -rw-r--r-- | R Scripts/predict-victims.R | 8 |
1 files changed, 5 insertions, 3 deletions
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() |