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)))

有没有办法使用插入符号包中的expand_grid函数在(0,1)之间找到TUNING_PARAMETER变量的最佳值。

Answer 1

是。您必须编写一个custom method来定义您自己的参数和模型。权重的代码需要放到fit模块中。