당신은 모델에 대한 모든 코드가 위치한 캐럿 GitHub의 페이지에 별도의 models에 대한 자세한 내용을 확인할 수 있습니다. 캐럿 문서는 here입니다.
그러나 당신이 2 개 개의 모델 사이에 사용되는 매개 변수의 차이에주의해야합니다
xgbLinear
사용 :
xgbTree
사용 nrounds, 람다, 알파, ETA : nrounds, MAX_DEPTH, ETA, 감마, colsample_bytree, min_child_weight
이러한 선택 사항은 모델 결과에 영향을 미치며 다른 예측 결과가 발생합니다. 따라서 정확도도 다릅니다. xgboost에서 사용할 수있는 다른 옵션은 xgboost의 기본 설정과 함께 사용됩니다.
캐럿 모델에는 dotdotdot (...) 옵션이 있습니다. 따라서 gamma
을 xgbLinear
으로 설정하려면 train 기능에서 지정할 수 있습니다. 그러나 그리드 매개 변수에는 없습니다. 다른 모든 xgboost 매개 변수에도 동일하게 적용됩니다.
(아주 나쁜) 예 :
grid = expand.grid(nrounds = c(10,20), lambda= c(0.1), alpha = c(1), eta = c(0.1))
train(Species ~ ., data = iris, method = "xgbLinear", tuneGrid = grid, gamma = 0.5)