我在插入符号包中的自定义训练模型遇到问题。我需要做一个 SVM 回归,我想找到 SVM 模型的所有参数 - 成本、西格玛和 epsilon。内置版本只有 cost 和 sigma。我已经在这里和这里找到了非常有用的提示,但我的模型仍然无法正常工作。
模型$grid(x = x, y = y, len = tuneLength, search = trControl$search) 中的错误:未使用的参数 (search = trControl$search)
这个错误是我得到的,我的代码在这里。
SVMrbf <- list(type = "Regression", library = "kernlab", loop = NULL)
prmrbf <- data.frame(parameters = data.frame(parameter = c('sigma', 'C', 'epsilon'),
class = c("numeric", "numeric", "numeric"),
label = c('Sigma', "Cost", "epsilon")))
SVMrbf$parameters <- prmrbf
svmGridrbf <- function(x, y, len = NULL) {
library(kernlab)
sigmas <- sigest(as.matrix(x), na.action = na.omit, scaled = TRUE, frac = 1)
expand.grid(sigma = mean(sigmas[-2]), epsilon = 10^(-5:0),
C = 2 ^(-5:len)) # len = tuneLength in train
}
SVMrbf$grid <- svmGridrbf
svmFitrbf <- function(x, y, wts, param, lev, last, weights, classProbs, ...) {
ksvm(x = as.matrix(x), y = y,
type = "eps-svr",
kernel = "rbfdot",
sigma = param$sigma,
C = param$C, epsilon = param$epsilon,
prob.model = classProbs,
...)
}
SVMrbf$fit <- svmFitrbf
svmPredrbf <- function(modelFit, newdata, preProc = NULL, submodels = NULL)
predict(modelFit, newdata)
SVMrbf$predict <- svmPredrbf
svmProb <- function(modelFit, newdata, preProc = NULL, submodels = NULL)
predict(modelFit, newdata, type="probabilities")
SVMrbf$prob <- svmProb
svmSortrbf <- function(x) x[order(x$C), ]
SVMrbf$sort <- svmSortrbf
svmRbfFit <- train(x = train.predictors1, y = train.response1, method = SVMrbf,
tuneLength = 10)
svmRbfFit
我找不到任何人,他们有同样的错误并且不知道实际出了什么问题。这段代码几乎就是我在网上找到的并稍作改动的东西。
顺便说一句,这是我的第一篇文章,所以希望它是可以理解的,如果不是,我可以添加更多信息。