我想为R中的'foo'类创建一个泛型函数predict(),以便当predict()的第二个参数属于类'foo'时调用它:
class(y.foo) <- "foo"
predict(x, y.foo)
是否有可能?怎么做?
试试这个。在这里,我们的foo
方法只是输出"foo"
,在实际代码中,它将被foo
您拥有的任何方法替换:
predict <- function(x, foo = NULL, ...) UseMethod("predict", foo)
predict.foo <- function(x, foo, ...) "foo" # replace with your foo method
predict.default <- function(x, foo = NULL, ...) if (is.null(foo))
stats::predict(x, ...) else stats::predict(x, foo, ...)
现在测试一下:
y.foo <- 1
class(y.foo) <- "foo"
predict(0, y.foo)
## [1] "foo"
fm <- lm(demand ~ Time, BOD)
predict(fm)
## 1 2 3 4 5 6
## 10.24286 11.96429 13.68571 15.40714 17.12857 20.57143
predict(fm, newdata = list(Time = 1:2))
## 1 2
## 10.24286 11.96429
predict(fm, list(Time = 1:2)) # same
## 1 2
## 10.24286 11.96429
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句