Convert demo to python (#9)

* convert demo to python * Update scratch/demo.qmd Co-authored-by: afmagee42 <[email protected]> --------- Co-authored-by: afmagee42 <[email protected]>
CDCgov · Dec 2, 2024 · 351aa89 · 351aa89
1 parent b55e8fb
commit 351aa89
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 25 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/scratch/demo.qmd b/scratch/demo.qmd
@@ -34,7 +34,7 @@ By definition, $v$ and $\lambda$ are an eigenvector and eigenvalue of $K$ if $Kv
 # do the eigenvalue analysis
 e = np.linalg.eig(K)
 # which eigenvalue is the dominant one?
-i = np.argmax(e.eigenvalues)
+i = np.argmax(np.abs(e.eigenvalues))
 dominant_eigenvalue = e.eigenvalues[i]
 # get the corresponding eigenvector
 v = e.eigenvectors[:, i]