diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2018-03-28 13:42:24 +0100 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2018-03-28 13:42:24 +0100 |
| commit | ae194ce592ece4e66b6ca7f3e1318ef7b9376dc3 (patch) | |
| tree | ea10c1d9849bc1010680cce19654366018a26fff /R/gensvm.grid.R | |
| parent | Add line stating refit is starting (diff) | |
| download | rgensvm-ae194ce592ece4e66b6ca7f3e1318ef7b9376dc3.tar.gz rgensvm-ae194ce592ece4e66b6ca7f3e1318ef7b9376dc3.zip | |
Properly handle factors for string parameters in gridsearch
Diffstat (limited to 'R/gensvm.grid.R')
| -rw-r--r-- | R/gensvm.grid.R | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/R/gensvm.grid.R b/R/gensvm.grid.R index 92478c2..3dad021 100644 --- a/R/gensvm.grid.R +++ b/R/gensvm.grid.R @@ -463,7 +463,11 @@ gensvm.cv.results <- function(results, param.grid, cv.idx, y.true, } for (parname in names(param.grid)) { - df[[sprintf("param.%s", parname)]][pidx] <- param[[parname]] + header <- sprintf("param.%s", parname) + val <- param[[parname]] + if (is.factor(val)) + val <- levels(val)[val] + df[[header]][pidx] <- val } j <- 1 |