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/gensvm.R | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'R/gensvm.R')

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)
-- 
cgit v1.2.3