r - 如何区分 R 中的函数

Question

我一直在尝试编写能够使用不同分类函数的代码。但是，参数因我使用的分类函数而异。我想要这样的东西:

classification_flow <- function(classification_function, ...) {
    if (classification_function == randomForest) {
        ...
    } 
    else if (classification_function == svm) {
        ...
    }
}

当然，这行不通，因为 == 不是为函数构建的。我试过使用 str、names、attr，并查看了一些 methods 和 UseMethod ，但我找不到合适的方法。

谁能帮帮我？

谢谢，
杰斯

PS:在这种特殊情况下，我要做的是获取概率矩阵作为输出，类似这样的事情:

classification_flow <- function(classification_function, train, classes, ...) {
    if (classification_function == randomForest) {
        mat = classification_function(train, classes, type="prob")
    } 
    else if (classification_function == svm) {
        mat = classification_function(train, classes, probabilities = T)
    }
    return(mat)
}

如果你知道更优雅的解决方案...

Answer 1

您正在寻找替代品:

f <- function(x, FUN) {
  if (substitute(FUN) == 'max') {
    print('Max invoked')
  }
  FUN(x)
}

> f(1:4, sum)
[1] 10
> f(1:4, max)
[1] "Max invoked"
[1] 4