]> nmode's Git Repositories - Rnaught/blobdiff - R/computeLL.R
Remove 'shinyBS' dependency
[Rnaught] / R / computeLL.R
index 0730399eafa19c24de5ce5fc82fe0ca5f83cc3ef..01c0d57b4ca8a1831627806fd5a4b56e85f804b4 100644 (file)
@@ -2,29 +2,25 @@
 #'
 #' 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.
+#' @param NT Vector of case counts.
+#' @param p Discretized version of the serial distribution.
+#' @param R0 Basic reproductive ratio.
+#'
+#' @return This function returns the log-likelihood at the input variables and parameters.
+#'
+#' @keywords internal
+computeLL <- function(p, NT, R0) {
+    k <- length(p)
+    TT <- length(NT) - 1
+    mu_t <- rep(0, TT)
 
-#' @export
-# 
-computeLL      <-      function(p, NT, R0){
+    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)
-       
 }
-
-