From 6529d39742e335039e3238b22c66eea8a69629ef Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Fri, 30 Mar 2018 17:25:47 +0100
Subject: set feature names for coef result

---
 R/coef.gensvm.R    | 9 ++++++---
 R/gensvm.R         | 3 ++-
 man/coef.gensvm.Rd | 4 ++--
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/R/coef.gensvm.R b/R/coef.gensvm.R
index 09febc2..60317aa 100644
--- a/R/coef.gensvm.R
+++ b/R/coef.gensvm.R
@@ -2,7 +2,7 @@
 #'
 #' @description Returns the model coefficients of the GenSVM object
 #'
-#' @param object a \code{gensvm} object
+#' @param fit a \code{gensvm} object
 #' @param \dots further arguments are ignored
 #'
 #' @return The coefficients of the GenSVM model. This is a matrix of size
@@ -35,7 +35,10 @@
 #' fit <- gensvm(x, y)
 #' V <- coef(fit)
 #'
-coef.gensvm <- function(object, ...)
+coef.gensvm <- function(fit, ...)
 {
-    return(object$V)
+    V <- fit$V
+    name <- c("translation", fit$feature.names)
+    rownames(V) <- name
+    return(V)
 }
diff --git a/R/gensvm.R b/R/gensvm.R
index cc30b4b..b70be78 100644
--- a/R/gensvm.R
+++ b/R/gensvm.R
@@ -182,7 +182,8 @@ gensvm <- function(X, y, p=1.0, lambda=1e-8, kappa=0.0, epsilon=1e-6,
                    classes = classes, V = out$V, n.iter = out$n.iter, 
                    n.support = out$n.support, 
                    training.time = out$training.time,
-                   X.train = if(kernel == 'linear') NULL else X)
+                   X.train = if(kernel == 'linear') NULL else X,
+                   feature.names = colnames(X))
     class(object) <- "gensvm"
 
     return(object)
diff --git a/man/coef.gensvm.Rd b/man/coef.gensvm.Rd
index cb6910b..0132ab9 100644
--- a/man/coef.gensvm.Rd
+++ b/man/coef.gensvm.Rd
@@ -4,10 +4,10 @@
 \alias{coef.gensvm}
 \title{Get the coefficients of the fitted GenSVM model}
 \usage{
-\method{coef}{gensvm}(object, ...)
+\method{coef}{gensvm}(fit, ...)
 }
 \arguments{
-\item{object}{a \code{gensvm} object}
+\item{fit}{a \code{gensvm} object}
 
 \item{\dots}{further arguments are ignored}
 }
-- 
cgit v1.2.3