R/CoefZOIP.R

In jucdiaz/ZOIP: ZOIP Distribution, ZOIP Regression, ZOIP Mixed Regression

#' coef.ZOIP
#'
#' Extract ZOIP model coefficients.
#'
#' @param object An object of class \code{ZOIP}.
#' @param ... other arguments.
#'
#' @examples
#'
#' #Test 1--------------------------------------------------
#' library(ZOIP)
#' library(boot)
#' n<-1000
#' x1<-stats::runif(n)
#' x2<-stats::runif(n)
#' b1<-0.3
#' b2<--0.5
#' b3<-0.9
#' sigma_i<-boot::inv.logit(b1+b2*x1+b3*x2)
#' c1<-0.2
#' c2<--1
#' c3<-0.1
#' mu_i<-boot::inv.logit(c1+c2*x1)
#' d1<-0.07
#' p0_i<-rep(d1,length(n))
#' e1<-0.02
#' e2<--4
#' p1_i<-boot::inv.logit(e1+e2*x2)
#' param<-cbind(mu_i,sigma_i,p0_i,p1_i)
#' system.time(y_i<-apply(param,1,function(x){rZOIP(1,mu=x[1],sigma=x[2],
#'                                                  p0=x[3],p1=x[4],family='R-S')}))
#' data<-as.data.frame(cbind(y_i,x1,x2))
#' formula.mu=y_i~x1
#' formula.sigma=~x1+x2
#' formula.p0=~1
#' formula.p1=~x1+x2
#' link=c('logit','logit','identity','logit')
#' family='R-S'
#' mod<-RM.ZOIP(formula.mu=formula.mu,formula.sigma=formula.sigma,
#'              formula.p0=formula.p0,formula.p1=formula.p1,data=data,link=link,family=family)
#' coef(mod)
#'
#'
#' @export

coef.ZOIP<-function(object, ...){

  object$nparm[object$Vec_Bool==T]<--1

  a<-cumsum(object$nparm)

  Aux<-c(0)
  names(Aux)<-c('(intercept)')

  if(object$Vec_Bool[1]==FALSE){elem.mu<-object$par[seq(1,a[1])]}else elem.mu<-Aux
  if(object$Vec_Bool[2]==FALSE){elem.sigma<-object$par[seq(a[1]+1,a[2])]}else elem.sigma<-Aux
  if(object$Vec_Bool[3]==FALSE){elem.p0<-object$par[seq(a[2]+1,a[3])]}else elem.p0<-Aux
  if(object$Vec_Bool[4]==FALSE){elem.p1<-object$par[seq(a[3]+1,a[4])]}else elem.p1<-Aux

  result<-list(Parameters.mu=NULL,Parameters.sigma=NULL,Parameters.p0=NULL,Parameters.p1=NULL)

  result$Parameters.mu<-elem.mu
  result$Parameters.sigma<-elem.sigma
  result$Parameters.p0<-elem.p0
  result$Parameters.p1<-elem.p1

  return(result)
}