R/logLikHess.R

Defines functions logLikHess

## Calculate the Hessian of the function, either by analytic or numeric method
logLikHess <- function( theta, fnOrig, gradOrig=NULL, hessOrig=NULL,
      start = NULL, fixed = NULL, gradAttr = NULL, hessAttr = NULL, ... ) {

   # argument "gradAttr" should be
   #    - FALSE if the gradient is not provided as attribute of the log-lik value
   #    - TRUE  if the gradient is provided as attribute of the log-lik value
   #    - NULL  if this is not known
   # argument "hessAttr" should be
   #    - FALSE if the Hessian is not provided as attribute of the log-lik value
   #    - TRUE  if the Hessian is provided as attribute of the log-lik value
   #    - NULL  if this is not known

   theta <- addFixedPar( theta = theta, start = start, fixed = fixed, ...)
   
   if(!is.null(hessOrig)) {
       hessian <- as.matrix(hessOrig( theta, ... ))
   } else {
      if( is.null( hessAttr ) || hessAttr || is.null( gradAttr ) ) {
         llVal <- fnOrig( theta, ... )
         gradient <- attr( llVal, "gradient" )
         hessian <- attr( llVal, "hessian" )
         gradAttr <- !is.null( gradient )
         hessAttr <- !is.null( hessian )
      }
      if( !hessAttr ) {
         if( !is.null( gradOrig ) ) {
            grad2 <- logLikGrad
         } else if( gradAttr ) {
            grad2 <- function( theta, fnOrig = NULL, gradOrig = NULL, ... ) {
               gradient <- attr( fnOrig( theta, ... ), "gradient" )
               gradient <- sumGradients( gradient, length( theta ) )
               return( gradient )
            }
         } else {
            grad2 <- NULL
         }
         hessian <- numericHessian( f = logLikFunc, grad = grad2, t0 = theta,
            fnOrig = fnOrig, gradOrig = gradOrig, ... )
      }
   }
   rownames( hessian ) <- colnames( hessian ) <- names( theta )

   if( !is.null( fixed ) ) {
      hessian <- hessian[ !fixed, !fixed, drop = FALSE ]
   }

   return( hessian )
}

Try the maxLik package in your browser

Any scripts or data that you put into this service are public.

maxLik documentation built on May 29, 2024, 2:32 a.m.