diff options
| author | Thibaut Horel <thibaut.horel@gmail.com> | 2015-07-02 22:37:45 -0700 |
|---|---|---|
| committer | Thibaut Horel <thibaut.horel@gmail.com> | 2015-07-02 22:37:45 -0700 |
| commit | 58ed50d980a1ba240bb50b27c42db0a679f00b43 (patch) | |
| tree | e7f410984ead0c8b2163edbae9d95dae66a7134b /R Scripts/find-parents.R | |
| parent | 5a76e2393e4f2d89f885ea99c473da840d0cd7db (diff) | |
| parent | 110069d77815a3d62e3526f18b2a34fb79beff1e (diff) | |
| download | criminal_cascades-58ed50d980a1ba240bb50b27c42db0a679f00b43.tar.gz | |
Merge branch 'master' of github.com:Thibauth/criminal_cascades
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 |