Передача оператора if в качестве аргумента функции в R

Question

Я пишу функцию с несколькими функциями внутри нее (обертка для обучения и тестирования нескольких моделей с пакетом каретки).

Проблема в том, что иногда я хотел бы использовать аргументы по умолчанию, и иногда я хотел бы указать свой собственный. В моем случае иногда я хочу изменить tuneGrid, и иногда я хочу использовать сетку по умолчанию. Я не могу использовать эллипсы (...), чтобы передать этот аргумент для этого, потому что каждая модель нуждается в отдельной сетке.

Как включить аргумент if в аргументы функций для этого? Например,

family = NULL 
y <- rnorm(1000); x <- rnorm(1000) 
glm(formula=y~x, if(!is.null(family)){family=family}, data=data.frame(x=x, y=y)) 

Это не работает по ряду причин, но я надеюсь, что вы получите эту идею. Я пытаюсь использовать что-то вроде этого, а не в качестве второго аргумента для объяснения запятых:

noquote(ifelse(!is.null(family), paste0(noquote("family="), family, "(),"))) 

Пожалуйста, дайте мне знать, если у вас есть какие-либо идеи.

Answer 1

Вы можете построить свой вызов функции в качестве цитируемого выражения и динамически изменить его. Что-то вроде

addGrid<-function(expr, grid=NULL) { 
    if(!is.null(grid)) { 
     ll<-as.list(expr) 
     ll$tuneGrid <- grid 
     as.call(ll) 
    } else { 
     expr 
    } 
} 

mycall<-quote(train(form= medv~., data = Boston, method = "glmboost")) 
addGrid(mycall) 
# train(form = medv ~ ., data = Boston, method = "glmboost") 
addGrid(mycall, quote(glmBoostGrid)) 
# train(form = medv ~ ., data = Boston, method = "glmboost", tuneGrid = glmBoostGrid) 

это возвращает выражение, которое будет затем вызвать eval() на выполнение.

В противном случае вы можете использовать if с частичным заявлением, чтобы, возможно, заполнить дополнительные значения.

mybase<-function(...) { 
    train(form= medv~., data = Boston, method = "glmboost", ...) 
} 
result <- if (!is.null(family)) { 
    mybase(family=family) 
} else { 
    mybase() 
}