R/mExWAS-extract.R

In rexposome: Exposome exploration and outcome data analysis

#' @describeIn mExWAS Method to obtain the generated model
#' @param object object of class \code{mExWAS}.
#' @param type (default \code{"test"}). Can take \code{"test"} or \code{"raw"}
#' to obtain a \code{data.frame} of \code{glmnet} result.
#' @param sort NOT USED
setMethod(
    f = "extract",
    signature = "mExWAS",
    definition = function(object, type = "test", sort = TRUE) {
        type <- match.arg(type, choices = c("test", "raw"))
        if(type == "test") {
            return(as.data.frame(as.matrix(stats::coef(object@result[[1]], c(object@result[[1]]$lambda.min, object@result[[1]]$lambda.1se)))))
        } else if(type == "raw") {
            return(object@result[[2]])
        }
    }
)