diff options
| author | Ben Green <ben@SEASITs-MacBook-Pro.local> | 2015-07-01 00:49:23 -0400 |
|---|---|---|
| committer | Ben Green <ben@SEASITs-MacBook-Pro.local> | 2015-07-01 00:49:23 -0400 |
| commit | 8e09ca6ca68c71bdab65525b529e2adfa281823c (patch) | |
| tree | d395bfcb0f9f0bc1092072ae1a8a9d3ad9c98a4a /R Scripts/find-parents.R | |
| parent | 6e527bbf612465bf5d739b9652abc0165550993c (diff) | |
| download | criminal_cascades-8e09ca6ca68c71bdab65525b529e2adfa281823c.tar.gz | |
Got predict-victims running in parallel, drastically reducing the time
for each test. Also changed how we get the rankings of infected
individuals each day.
Diffstat (limited to 'R Scripts/find-parents.R')
| -rw-r--r-- | R Scripts/find-parents.R | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/R Scripts/find-parents.R b/R Scripts/find-parents.R index 3ec8809..023d7ba 100644 --- a/R Scripts/find-parents.R +++ b/R Scripts/find-parents.R @@ -6,8 +6,8 @@ # source('criminal_cascades/R Scripts/structural.R') ##### Initialize parameters based on what ml2 found -alpha = 0.061 -delta = 0.082 +alpha = 0.18 +delta = 0.09 ##### Get weights edges = dag_dat_test[!is.na(dag_dat_test$t2),] @@ -21,20 +21,27 @@ weights = p/p_tilde edges$weight = weights ##### Find most likely parents -parents = data.frame(vic=0,Npars=0,par_rank=0) +parents = data.frame(vic=0,Npars=0,par_rank=0,rand_rank=0) vics = setdiff(vic_ids,seeds) +print(length(vics)) for (u in vics){ + if(which(vics==u) %% 500 == 0) print(which(vics==u)) u_parents = edges[edges$to==u,] u_parents = u_parents[order(u_parents$weight,decreasing=T),] Nparents = dim(u_parents)[1] infector = V(g)$infector[u] infectorID = which(u_parents$from==infector) - parents[which(vics==u),] = c(u, Nparents, infectorID) + randID = sample(1:Nparents,1) + parents[which(vics==u),] = c(u, Nparents, infectorID, randID) } ##### Get some summary statistics on how well +mean(parents$par_rank==1) median(parents$par_rank[parents$Npars>9]) median(parents$par_rank[parents$Npars>99]) +edges[edges$to==2847,] +## baseline alg +# for each vic, find potential parents, pick one at random |