getGmodel = function(family, task) {
if (family == "tree") {
function(dataHO, targetHO, param) {
- require(rpart)
+ base::require(rpart)
method <- ifelse(task == "classification", "class", "anova")
if (is.null(colnames(dataHO)))
colnames(dataHO) <- paste0("V", 1:ncol(dataHO))
else if (family == "knn") {
if (task == "classification") {
function(dataHO, targetHO, param) {
- require(class)
+ base::require(class)
function(X) class::knn(dataHO, X, cl=targetHO, k=param)
}
}
else {
function(dataHO, targetHO, param) {
- require(FNN)
+ base::require(FNN)
function(X) FNN::knn.reg(dataHO, X, y=targetHO, k=param)$pred
}
}
getParams = function(family, data, target, task) {
if (family == "tree") {
# Run rpart once to obtain a CV grid for parameter cp
- require(rpart)
+ base::require(rpart)
df <- data.frame(cbind(data, target=target))
ctrl <- list(
cp = 0,