aboutsummaryrefslogtreecommitdiff
path: root/R
diff options
context:
space:
mode:
authorGertjan van den Burg <gertjanvandenburg@gmail.com>2018-04-03 19:41:35 -0400
committerGertjan van den Burg <gertjanvandenburg@gmail.com>2018-04-03 19:41:35 -0400
commitb7a63527991cce68daa984f36088f23cf7d1a199 (patch)
treeed995adb80a38853faa50761f8f0b85ba32774e9 /R
parentRemove todo item from wrapper (diff)
downloadrgensvm-b7a63527991cce68daa984f36088f23cf7d1a199.tar.gz
rgensvm-b7a63527991cce68daa984f36088f23cf7d1a199.zip
fix for weights in refit
Diffstat (limited to 'R')
-rw-r--r--R/gensvm.R2
-rw-r--r--R/gensvm.grid.R5
2 files changed, 6 insertions, 1 deletions
diff --git a/R/gensvm.R b/R/gensvm.R
index 5652a4c..d59c7e0 100644
--- a/R/gensvm.R
+++ b/R/gensvm.R
@@ -169,7 +169,7 @@ gensvm <- function(x, y, p=1.0, lambda=1e-8, kappa=0.0, epsilon=1e-6,
lambda,
kappa,
epsilon,
- weight.idx,
+ as.integer(weight.idx),
raw.weights,
as.integer(kernel.idx),
gamma,
diff --git a/R/gensvm.grid.R b/R/gensvm.grid.R
index d4b06b5..62add19 100644
--- a/R/gensvm.grid.R
+++ b/R/gensvm.grid.R
@@ -225,6 +225,11 @@ gensvm.grid <- function(x, y, param.grid='tiny', refit=TRUE, scoring=NULL, cv=3,
if (refit && !is.na(best.index)) {
gensvm.args <- as.list(best.params)
+ # Stupid factors...
+ if ("weights" %in% names(gensvm.args))
+ gensvm.args$weights <- as.character(gensvm.args$weights)
+ if ("kernel" %in% names(gensvm.args))
+ gensvm.args$kernel <- as.character(gensvm.args$kernel)
gensvm.args$x <- x
gensvm.args$y <- y
gensvm.args$verbose <- if(verbose>1) 1 else 0