R/utils_pmax.R

In snaketron/IgGeneUsage: Differential gene usage in immune repertoires

# Description:
# compute pmax -> probability of DGU
get_pmax_dgu <- function(glm_ext) {
  
  get_pmax_gene <- function(x, beta_data) {
    p <- beta_data[,x]
    l <- length(p)
    o <- 2*max(sum(p<=0)/l, sum(p>=0)/l)-1
    return(o)
  }
  
  beta_data <- glm_ext$beta_gene_mu
  pmax <- vapply(X = seq_len(length.out = ncol(beta_data)),
                 FUN = get_pmax_gene,
                 beta_data = beta_data,
                 FUN.VALUE = numeric(length = 1))
  pmax <- 2 * pmax - 1
  return(pmax)
}

# Description:
# compute pmax -> from numeric vector
get_pmax <- function(x) {
  p <- x
  l <- length(p)
  o <- 2*max(sum(p <=0)/l, sum(p>=0)/l)-1
  return(o)
}