Remove exposed seed from pop

mrc-ide · Apr 22, 2020 · 6f4f9fe · 6f4f9fe
1 parent 4a6c18c
commit 6f4f9fe
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/index.js b/src/index.js
@@ -58,9 +58,12 @@ export const runModel = function(
   const model = Object.values(odin)[0];
   const nGroups = population.length;
 
+  // Remove the seed exposed population
+  const true_pop = population.map((p, i) => { return p - pars.E1_0[i] });
+
   const user = {
     ...pars,
-    S_0: population,
+    S_0: true_pop,
     tt_matrix: ttMatrix,
     mix_mat_set: transpose312(
       reshape3d(mixMatSet, [nGroups, nGroups, ttMatrix.length])