From 6e527bbf612465bf5d739b9652abc0165550993c Mon Sep 17 00:00:00 2001 From: Ben Green Date: Sun, 28 Jun 2015 17:38:33 -0400 Subject: Worked on synthetic data recovery so we can tell how high the actual infector is ranked among all potential parents. Cleaned up code for the predicting victims benchmarking test. --- R Scripts/generate-network.R | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'R Scripts/generate-network.R') diff --git a/R Scripts/generate-network.R b/R Scripts/generate-network.R index dc4a4f8..3b40969 100644 --- a/R Scripts/generate-network.R +++ b/R Scripts/generate-network.R @@ -9,7 +9,6 @@ delta = 0.15 # lmbda = 1/10 t_max = 1000 -# g = watts.strogatz.game(1, 100, 3, 0.25) N = 5000 g = forest.fire.game(nodes=N, fw.prob=0.3, ambs=1, directed=F) plot(g, vertex.size=5, vertex.label=NA) @@ -31,7 +30,7 @@ for (day in 1:t_max){ infected = setdiff(infected,seeds) # don't try to infect seeds inf.days = day + ceiling(alpha*rexp(length(infected),alpha)) V(g)$vic[infected] = TRUE - infects = (inf.days < V(g)$vic.day[infected]) %in% c(NA,T) + infects = (inf.days <= V(g)$vic.day[infected]) %in% c(NA,T) V(g)$vic.day[infected[infects]] = inf.days[infects] V(g)$infector[infected[infects]] = vic } @@ -76,3 +75,5 @@ infectors = cbind(setdiff(vic_ids,seeds), V(g)$infector[setdiff(vic_ids,seeds)], recovered$infector[recovered$victim %in% setdiff(vic_ids,seeds)]) mean(infectors[,2]==infectors[,3]) + +dag_dat_test[dag_dat_test$to==4984,] \ No newline at end of file -- cgit v1.2.3-70-g09d2