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