diff options
Diffstat (limited to 'R Scripts/fit-background.R')
| -rw-r--r-- | R Scripts/fit-background.R | 10 |
1 files changed, 5 insertions, 5 deletions
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") |