#' Classify Spatially Variable Genes to interpretable fitting classes
#'
#' Compare model fits with different models, using the
#' [**SpatialDE**](https://github.com/Teichlab/SpatialDE) Python package.
#'
#' @param x A numeric `matrix` of counts where genes are rows and cells are columns.
#'
#' Alternatively, a \linkS4class{SpatialExperiment} object.
#'
#' @param ... For the generic, arguments to pass to specific methods.
#' @param de_results `data.frame` resulting from [run()] or [spatialDE()].
#' @param coordinates A `data.frame` with sample coordinates. Each row is a
#' sample, the columns with coordinates should be named 'x' and 'y'.
#'
#' For the *SpatialExperiment* method, coordinates are taken from
#' `spatialCoords(x)`.
#'
#' @param qval_thresh `numeric` scalar, specifying the q-value significance
#' threshold to filter `de_results`. Only rows in `de_results` with
#' `qval < qval_thresh` will be kept. To disable, set `qval_thresh = NULL`.
#' @param assay_type A `character` string specifying the assay from `x` to use
#' as input. Defaults to `"counts"`.
#' @param verbose A `logical` controlling the display of a progress bar from the
#' Python package.
#'
#' @return `data.frame` of model_search results.
#'
#' @examples
#' ## Mock up a SpatialExperiment object wit 400 cells and 3 genes
#' set.seed(42)
#' spe <- mockSVG(size = 20, tot_genes = 3, de_genes = 1, return_SPE = TRUE)
#'
#' ## Run spatialDE with S4 integration
#' de_results <- spatialDE(spe)
#'
#' ## Run model search
#' model_search <- modelSearch(spe, de_results = de_results,
#' qval_thresh = NULL, verbose = FALSE
#' )
#'
#' @seealso
#' The individual steps performed by this function: [stabilize()],
#' [regress_out()] and [model_search()].
#'
#' @references
#' Svensson, V., Teichmann, S. & Stegle, O. SpatialDE: identification of
#' spatially variable genes. Nat Methods 15, 343–346 (2018).
#' \url{https://doi.org/10.1038/nmeth.4636}
#'
#' [**SpatialDE 1.1.3**](https://pypi.org/project/SpatialDE/1.1.3/): the version
#' of the Python package used under the hood.
#'
#' @author Davide Corso, Milan Malfait, Lambda Moses
#' @name modelSearch
NULL
#' @importFrom Matrix colSums
#' @importFrom checkmate assert_data_frame assert_number assert_flag
.modelSearch <- function(x, de_results, coordinates,
qval_thresh = 0.05, verbose = FALSE) {
assert_data_frame(de_results, all.missing = FALSE)
assert_number(qval_thresh, null.ok = TRUE)
assert_flag(verbose)
## Filter de_results
if (!is.null(qval_thresh)) {
de_results <- .filter_de_results(
de_results = de_results, qval_thresh = qval_thresh
)
}
sample_info <- data.frame(coordinates, total_counts = colSums(x))
.run_model_search(x = x, sample_info = sample_info, de_results = de_results,
verbose = verbose
)
}
.run_model_search <- function(x, sample_info, de_results, verbose = FALSE) {
proc <- basiliskStart(spatialDE_env, testload="scipy.optimize")
# Normalization
## Stabilize
.naiveDE_stabilize(proc, x)
stabilized <- basiliskRun(proc, function(store) {
as.matrix(store$stabilized)
}, persist=TRUE)
## Regress_out
.naiveDE_regress_out(proc, stabilized, sample_info)
regressed <- basiliskRun(proc, function(store) {
as.matrix(store$regressed)
}, persist=TRUE)
coordinates <- sample_info[, c("x", "y")]
.importPyModule(proc, !verbose)
.spatialDE_model_search(proc, regressed, coordinates, de_results)
out <- basiliskRun(proc, function(store) {
store$model_search
}, persist=TRUE)
out
}
#' @import methods
#' @export
#' @rdname modelSearch
setGeneric("modelSearch",
function(x, de_results, ...) standardGeneric("modelSearch"),
signature = "x"
)
#' @export
#' @rdname modelSearch
setMethod("modelSearch", "matrix", .modelSearch)
#' @export
#' @rdname modelSearch
#' @importFrom SummarizedExperiment assay
#' @importFrom SpatialExperiment spatialCoords spatialCoordsNames<-
setMethod("modelSearch", "SpatialExperiment",
function(x, de_results, assay_type = "counts",
qval_thresh = 0.05, verbose = FALSE) {
## Rename spatialCoords columns to "x", "y"
spatialCoordsNames(x) <- c("x", "y")
coordinates <- as.data.frame(spatialCoords(x))
.modelSearch(
x = assay(x, assay_type), de_results = de_results,
coordinates = coordinates, qval_thresh = qval_thresh,
verbose = verbose
)
}
)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.