R/rpart.gamma.R

In henckr/distRforest: Distribution-based Random Forest

rpart.gamma <- function(y, offset, parms, wt)
{
  if (any(y <= 0)) stop("Response variable must be > 0, support of the gamma distribution is strictly positive")
  
    if (!is.null(offset)) y <- y - offset
    list(y = y, parms = NULL, numresp = 1L, numy = 1L,
	 summary = function(yval, dev, wt, ylevel, digits ) {
	     paste0("  mean=", formatg(yval, digits),
                    ", MSE=" , formatg(dev/wt, digits))
         },
	 text = function(yval, dev, wt, ylevel, digits, n, use.n ) {
	     if (use.n) paste0(formatg(yval, digits), "\nn=", n) else
             formatg(yval, digits)
         })
}