gamSpline Caret 包

Question

如何為我的樣條選擇最佳 df（自由度）。 我使用泊松回歸和樣條曲線來幫助我調整非線性變化。 使用 caret 包，使用 method = gamSpline 訓練函數，該函數測試僅 3 df。

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