From 14ec98c9bfc49f963e30159dc7db9f554088fb44 Mon Sep 17 00:00:00 2001 From: hannajankowski Date: Thu, 9 Jun 2022 17:57:31 -0400 Subject: Testing testing 123 --- R/computeLL.R | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 R/computeLL.R (limited to 'R/computeLL.R') diff --git a/R/computeLL.R b/R/computeLL.R new file mode 100644 index 0000000..0730399 --- /dev/null +++ b/R/computeLL.R @@ -0,0 +1,30 @@ +#' WP method background function computeLL +#' +#' This is a background/internal function called by \code{WP}. It computes the log-likelihood. +#' +#' @param NT vector of case counts +#' @param p discretized version of the serial distribution +#' @param R0 basic reproductive ratio +#' @return The function returns the variable \code{LL} which is the log-likelihood at the input variables and parameters. + +#' @export +# +computeLL <- function(p, NT, R0){ + + k <- length(p) + TT <- length(NT)-1 + mu_t <- rep(0, TT) + for(i in 1:TT){ + Nt <- NT[i:max(1,i-k+1)] +# print(Nt) +# print(p[1:min(k,i)]) + mu_t[i] <- sum(p[1:min(k,i)]*Nt) + } + mu_t <- R0*mu_t + LL <- -sum(mu_t)+sum(NT[-1]*log(mu_t)) + + return(LL) + +} + + -- cgit v1.2.3