From 9c1a5668803e735f034700c55028ffc0146f1e93 Mon Sep 17 00:00:00 2001 From: Naeem Model Date: Wed, 21 Jun 2023 05:38:42 +0000 Subject: Fix code formatting and remove unnecessary comments --- R/computeLL.R | 31 +++++++++++++------------------ 1 file changed, 13 insertions(+), 18 deletions(-) (limited to 'R/computeLL.R') diff --git a/R/computeLL.R b/R/computeLL.R index 0730399..6acfc78 100644 --- a/R/computeLL.R +++ b/R/computeLL.R @@ -3,28 +3,23 @@ #' 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 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){ +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)] + mu_t[i] <- sum(p[1:min(k, i)] * Nt) + } - 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)) + mu_t <- R0 * mu_t + LL <- sum(NT[-1] * log(mu_t)) - sum(mu_t) return(LL) - } - - -- cgit v1.2.3