From 717e7f2edda52100406d17826bd617b7915c5daf Mon Sep 17 00:00:00 2001 From: Ben Green Date: Wed, 9 Sep 2015 20:27:12 -0400 Subject: updated R scripts --- R Scripts/fit-background.R | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'R Scripts/fit-background.R') diff --git a/R Scripts/fit-background.R b/R Scripts/fit-background.R index d385e7f..d1799cd 100644 --- a/R Scripts/fit-background.R +++ b/R Scripts/fit-background.R @@ -17,16 +17,16 @@ counts = as.vector(t) infs = data.frame(days,counts) # define background function -fit = function(x, mu, A, phi) {mu + A*(sin((pi/365)*x+phi))^2} -fit_form = counts ~ mu + A*(sin((pi/365)*days+phi))^2 +fit = function(x, lambda, A, phi) {lambda + A*(sin((pi/365)*x+phi))^2} +fit_form = counts ~ lambda + A*(sin((pi/365)*days+phi))^2 # explore data plot(days,counts) -curve(fit(x, mu=1, A=12, phi=2.8), add=TRUE ,lwd=2, col="steelblue") +curve(fit(x, lambda=1, A=12, phi=2.8), add=TRUE ,lwd=2, col="steelblue") -res = nls(formula=fit_form, data=infs, start=list(mu=1, A=12, phi=2.8)) +res = nls(formula=fit_form, data=infs, start=list(lambda=1, A=12, phi=2.8)) co = coef(res); co plot(t) plot(days,counts,pch=20,cex=0.4,xlab='Day',ylab='Number of Infections') -curve(fit(x, mu=co["mu"], A=co["A"], phi=co["phi"]), add=TRUE ,lwd=5, col="steelblue") +curve(fit(x, lambda=co["lambda"], A=co["A"], phi=co["phi"]), add=TRUE ,lwd=5, col="steelblue") -- cgit v1.2.3-70-g09d2