From 904e559845f48f38c7eb928d65876e3a064a17c7 Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Wed, 10 Feb 2016 15:27:35 -0500
Subject: add documentation, unify implementations of path and fit

---
 R/plot.sparsestep.R | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 R/plot.sparsestep.R

(limited to 'R/plot.sparsestep.R')

diff --git a/R/plot.sparsestep.R b/R/plot.sparsestep.R
new file mode 100644
index 0000000..87ab661
--- /dev/null
+++ b/R/plot.sparsestep.R
@@ -0,0 +1,30 @@
+#' @title Plot the SparseStep path
+#'
+#' @description Plot the coefficients of the SparseStep path
+#'
+#' @param obj a \code{sparsestep} object
+#' @param type the plot type, default: "s".
+#' @param lty line type, default: 1
+#' @param ... further argument to matplot
+#'
+#' @author
+#' Gertjan van den Burg (author and maintainer).
+#'
+#' @export
+#' @aliases plot
+#'
+#' @examples
+#' data(diabetes)
+#' attach(diabetes)
+#' fit <- sparsestep(x, y)
+#' plot(fit)
+#' pth <- sparsestep.path(x, y)
+#' plot(pth)
+plot.sparsestep <- function(obj, type="s", lty=1, ...)
+{
+	index <- log(obj$lambda)
+	coefs <- t(as.matrix(obj$beta))
+
+	matplot(index, coefs, xlab="Log lambda", ylab="Coefficients",
+	       	type=type, lty=lty, ...)
+}
-- 
cgit v1.2.3