2015-10-19 181 views
9

我正在嘗試用於預測的XGBoost技術。由於我的因變量是連續的,我正在使用XGBoost進行迴歸,但大多數可用於各種門戶的參考都是用於分類的。雖然我知道通過使用如何在R中使用XGBoost算法進行迴歸?

objective = "reg:linear" 

我們可以做迴歸,但仍然需要一些其他參數的清晰度。如果有人能爲我提供一個R代碼段,這將是一個很大的幫助。

+0

@ Amarjeet:實際上,你收到一個錯誤,當您嘗試運行它或者是你的問題,而該如何調整參數? –

+0

@Amarjeet關於此主題的任何更新?我也有同樣的問題。 – deltascience

回答

5
xgboost(data = X, 
     booster = "gbtree", 
     objective = "binary:logistic", 
     max.depth = 5, 
     eta = 0.5, 
     nthread = 2, 
     nround = 2, 
     min_child_weight = 1, 
     subsample = 0.5, 
     colsample_bytree = 1, 
     num_parallel_tree = 1) 

這些都是您可以在使用樹型助推器時玩的所有參數。對於線性助推器,你可以使用下面的參數一起玩......

xgboost(data = X, 
     booster = "gblinear", 
     objective = "binary:logistic", 
     max.depth = 5, 
     nround = 2, 
     lambda = 0, 
     lambda_bias = 0, 
     alpha = 0) 

您可以參考xg.train()對這些參數的詳細含義xgboost CRAN文檔中的描述。

+0

我知道這是一個非常寬泛的問題,但是如果有任何與迴歸有關的具體答案將有助於理解。 – Amarjeet

+0

線性迴歸和二元邏輯迴歸是xgboost軟件包中最常用的方法... xgboost中的所有參數僅用於操作算法的提升部分......包中沒有太多的範圍來操縱迴歸技術無論是ols迴歸還是二元邏輯... – Gaurav