aboutsummaryrefslogtreecommitdiff
path: root/R/gensvm.R
diff options
context:
space:
mode:
authorGertjan van den Burg <gertjanvandenburg@gmail.com>2018-03-30 17:25:47 +0100
committerGertjan van den Burg <gertjanvandenburg@gmail.com>2018-03-30 17:25:47 +0100
commit6529d39742e335039e3238b22c66eea8a69629ef (patch)
tree2cd5c5563ec234042c3f48df40160cc390889c73 /R/gensvm.R
parentAdd plot function for binary data (diff)
downloadrgensvm-6529d39742e335039e3238b22c66eea8a69629ef.tar.gz
rgensvm-6529d39742e335039e3238b22c66eea8a69629ef.zip
set feature names for coef result
Diffstat (limited to 'R/gensvm.R')
-rw-r--r--R/gensvm.R3
1 files changed, 2 insertions, 1 deletions
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)