documentation update

This commit is contained in:
hetong007
2015-01-20 16:24:01 -08:00
parent d87cb24793
commit 42110f3d70
2 changed files with 2 additions and 2 deletions

View File

@@ -51,7 +51,7 @@ train <- agaricus.train
bst <- xgboost(data = train$data, label = train$label, max.depth = 2,
eta = 1, nround = 2,objective = "binary:logistic")
xgb.dump(bst, 'xgb.model.dump', with.stats = T)
xgb.dump(bst, 'xgb.model.dump', with.stats = TRUE)
#agaricus.test$data@Dimnames[[2]] represents the column names of the sparse matrix.
xgb.model.dt.tree(agaricus.train$data@Dimnames[[2]], filename_dump = 'xgb.model.dump')