R/mean-diff-continuous.R

In bayesMeanScale: Bayesian Post-Estimation on the Mean Scale

meanDiffContinuousF <- function(pred_start, pred_end, model_data, marg_list, at, at_means, i, h){

  if(!is.null(at) & at_means==F){

    atVars <- names(expand.grid(at))

    predDiffOrg <- ((pred_end - pred_start)/h) %>%
      data.table::as.data.table()

    predDiff <- predDiffOrg %>%
      cbind(model_data[, ..atVars]) %>%
      .[, lapply(.SD, mean), by=atVars, .SDcols=!atVars] %>%
      data.table::melt(id.vars       = atVars,
           variable.name = 'which_diff',
           value.name    = 'diff') %>%
      .[, !"which_diff"] %>%
      .[, `:=`(comparison  = "Instantaneous rate of change",
               marg_effect = marg_list$marg[[i]])]

  }
  
  if(is.null(at) & at_means==F){

    predDiff <- data.table::data.table(diff        = colMeans(((pred_end - pred_start)/h)),
                                       comparison  = "Instantaneous rate of change",
                                       marg_effect = marg_list$marg[[i]])

  }
  
  if(!is.null(at) & at_means==T){
    
    atVars   <- names(expand.grid(at))
    atValues <- expand.grid(at)
    
    predDiffOrg <- ((pred_end - pred_start)/h) %>%
      data.table::as.data.table()
    
    predDiff <- predDiffOrg %>%
      cbind(atValues) %>%
      data.table::melt(id.vars       = atVars,
           variable.name = 'which_diff',
           value.name    = 'diff') %>%
      .[, !"which_diff"] %>%
      .[, `:=`(comparison  = "Instantaneous rate of change",
               marg_effect = marg_list$marg[[i]])]
    
  }
  
  if(is.null(at) & at_means==T){
    
    predDiff <- data.table(diff        = colMeans(((pred_end - pred_start)/h)),
                           comparison  = "Instantaneous rate of change",
                           marg_effect = marg_list$marg[[i]])
    
  }

  return(predDiff)

}