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

TST: Added test for xgb.plot.tree

Browse Source
This commit is contained in:
terrytangyuan 2015-09-07 21:49:27 -04:00
parent d833038ba1
commit 408c3a62a8
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
R-package/tests/testthat/test_helpers.R
View File

@ -25,3 +25,7 @@ test_that("xgb.importance works", {
xgb.dump(bst, 'xgb.model.dump', with.stats = T) xgb.dump(bst, 'xgb.model.dump', with.stats = T)
importance <- xgb.importance(sparse_matrix@Dimnames[[2]], 'xgb.model.dump') importance <- xgb.importance(sparse_matrix@Dimnames[[2]], 'xgb.model.dump')
}) })
test_that("xgb.plot.tree works", {
xgb.plot.tree(agaricus.train$data@Dimnames[[2]], model = bst)
})