gamSpline Caret包

Question

如何为我的样条选择最优的df(自由度)。我使用泊松回归和样条来帮助我调整非线性变化。使用插入符号包，使用method = gamSpline训练函数测试只有3df的函数。

model <- train(
  RBC ~ elapsed, 
  obgyn_aleph,
  method = "gamSpline",
  trControl = trainControl(
    method = "cv", 
    number = 10,
    verboseIter = TRUE
  )
)

聚合结果选择调整参数在全训练集上拟合df =3

这是默认的吗？如果是这样，我如何才能更改它？

丹尼尔·Tnx

Answer 1

tuneGrid参数允许用户指定调整参数的自定义网格，在本例中为df

 model <- train(
      RBC ~ elapsed, 
      obgyn_aleph,
      method = "gamSpline",
      trControl = trainControl(
        method = "cv", 
        number = 10,
        verboseIter = TRUE
      ),
      tuneGrid = data.frame(df=seq(2,20,by=2))
    )