just some minor tests with ml

1 files changed, 9 insertions, 7 deletions

diff --git a/R Scripts/generate-network.R b/R Scripts/generate-network.R
index dab81a4..db7012d 100644
--- a/R Scripts/generate-network.R
+++ b/R Scripts/generate-network.R
@@ -3,9 +3,9 @@ setwd("~/Documents/Cascade Project/")
 source('criminal_cascades/R Scripts/temporal.R')
 source('criminal_cascades/R Scripts/structural.R')
 
-alpha = 1/10
-beta = 0.01
-delta = 0.25
+alpha = 0.1
+beta = 0.25
+delta = 0.1
 # lmbda = 1/10
 t_max = 1000
 
@@ -16,7 +16,7 @@ plot(g, vertex.size=3, vertex.label=NA)
 V(g)$seed = runif(vcount(g))<beta
 seeds = which(V(g)$seed)
 V(g)$vic = V(g)$seed
-V(g)$vic.day[V(g)$seed] = 1#sample(1:t_max, sum(V(g)$seed))
+V(g)$vic.day[V(g)$seed] = sample(1:t_max, sum(V(g)$seed), replace=T) #1 if testing ml2
 V(g)$spawn.date = 0
 V(g)$infector = NA
 
@@ -31,9 +31,11 @@ for (day in 1:t_max){
     inf.days = day + ceiling(rexp(length(infected),alpha))
     realized = ((inf.days <= V(g)$vic.day[infected]) %in% c(NA,T)) & (inf.days<=t_max)
     infected = infected[realized]
-    V(g)$vic[infected] = TRUE
-    V(g)$vic.day[infected] = inf.days[realized]
-    V(g)$infector[infected] = vic
+    if(sum(realized)){
+      V(g)$vic[infected] = TRUE
+      V(g)$vic.day[infected] = inf.days[realized]
+      V(g)$infector[infected] = vic
+    }
   }
 }