Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Plot model deepness

Browse Source 
New function to explore the model by ploting the way splits are done.
This commit is contained in:
pommedeterresautee
2015-11-24 11:45:32 +01:00
parent 60dd75745f
commit 485b30027f
12 changed files with 383 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
R-package/man/xgb.plot.importance.Rd
View File

@@ -25,12 +25,12 @@ In particular you may want to override the title of the graph. To do so, add \co
\examples{
data(agaricus.train, package='xgboost')
#Both dataset are list with two items, a sparse matrix and labels
#(labels = outcome column which will be learned).
#Both dataset are list with two items, a sparse matrix and labels
#(labels = outcome column which will be learned).
#Each column of the sparse Matrix is a feature in one hot encoding format.
train <- agaricus.train
bst <- xgboost(data = train$data, label = train$label, max.depth = 2,
bst <- xgboost(data = train$data, label = train$label, max.depth = 2,
eta = 1, nthread = 2, nround = 2,objective = "binary:logistic")
#train$data@Dimnames[[2]] represents the column names of the sparse matrix.