updates

CDCgov · Nov 26, 2024 · 1b4a959 · 1b4a959
1 parent c403f16
commit 1b4a959
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/scratch/ngm-structured-sir.R b/scratch/ngm-structured-sir.R
@@ -8,8 +8,8 @@ ngm_sir <- function(N, V, K, VE, p_severe) {
   K <- K * S / N
 
   eigenvalues <- eigen(K)
-  r_effective <- max(eigenvalues$values)
-  infections <- eigenvalues$vectors[, which.max(eigenvalues$values)] * N
+  r_effective <- max(abs(eigenvalues$values))
+  infections <- eigenvalues$vectors[, which.max(abs(eigenvalues$values))]
   severe_infections <- infections * p_severe
   return(list(r_e = r_effective, infections = infections, severe_infections = severe_infections))
 }
@@ -22,8 +22,6 @@ K <- matrix(c(
   1, 1, 1, 1 # general
 ), nrow = 4, ncol = 4)
 
-
-
 N <- c(100, 100, 10, 790) # pop size: core, kids, travelers, general
 V <- c(10, 0, 0, 0) # doses
 VE <- 0.7 # vaccine efficacy