R/get_mu.R

In mohuangx/SAVER: Single-Cell RNA-Seq Gene Expression Recovery

#' Output prior predictions
#' 
#' Outputs prior predictions
#' 
#' This function outputs prior mean predictions \eqn{\mu} used in fitting the 
#' SAVER model.
#' 
#' @param x Original count matrix.
#' 
#' @param saver.obj SAVER output.
#' 
#' @return A matrix of prior mean predictions
#' 
#' @examples 
#' data("linnarsson")
#' data("linnarsson_saver")
#' 
#' mu <- get.mu(linnarsson, linnarsson_saver)
#' 
#' @export

get.mu <- function(x, saver.obj) {
  mu <- (saver.obj$estimate^2/saver.obj$se^2 - x)/
    sweep(saver.obj$estimate/saver.obj$se^2, 2, saver.obj$info$size.factor, "-")
  round(mu, 3)
}