From 904e559845f48f38c7eb928d65876e3a064a17c7 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Wed, 10 Feb 2016 15:27:35 -0500 Subject: add documentation, unify implementations of path and fit --- R/coef.sparsestep.R | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 R/coef.sparsestep.R (limited to 'R/coef.sparsestep.R') diff --git a/R/coef.sparsestep.R b/R/coef.sparsestep.R new file mode 100644 index 0000000..3a6286c --- /dev/null +++ b/R/coef.sparsestep.R @@ -0,0 +1,30 @@ +#' @title Get the coefficients of a fitted SparseStep model +#' +#' @description Returns the coefficients of the SparseStep model. +#' +#' @param obj a "sparsestep" object +#' +#' @return The coefficients of the SparseStep model (i.e. the betas). If the +#' model was fitted with an intercept this will be the first value in the +#' resulting vector. +#' +#' @export +#' @aliases coef +#' +#' @examples +#' x <- matrix(rnorm(100*20), 100, 20) +#' y <- rnorm(100) +#' fit <- sparsestep(x, y) +#' coef(fit) +#' +coef.sparsestep <- function(obj, ...) +{ + if (obj$intercept) { + beta <- rbind(obj$a0, obj$beta) + } else { + beta <- obj$beta + } + nbeta <- drop0(Matrix(as.matrix(beta))) + + return(nbeta) +} -- cgit v1.2.3