R中的自定义调整网格插入符包

Question

我有一个带有加权因子的线性模型，该加权因子对最近的观察结果进行加权。权重使用我想使用调整网格优化的调整参数。下面是一个简单的例子：

require(data.table)
require(caret)
SMOOTHING_PARAMETER <- 0.2
dt <- data.table(y = rnorm(10),
                 x = rnorm(10))

model <- train(y ~ x,
               data = dt,
               method = "lm",
               weights = (1 + SMOOTHING_PARAMETER) ^(1:nrow(dt)))

有没有什么方法可以使用 caret 包中的expand_grid 函数来找到 (0,1) 之间的TUNING_PARAMETER 变量的最佳值。

Answer 1

是的。你必须写一个custom method来定义你自己的参数和模型。权重的代码需要进入fit 模块。