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

TST: Added one minor check for xgb.importance

Browse Source
This commit is contained in:
terrytangyuan 2015-09-07 22:51:14 -04:00
parent fbf2a5feed
commit 33f1ab3ae1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -24,6 +24,7 @@ test_that("xgb.dump works", {
test_that("xgb.importance works", { 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')
expect_equal(dim(importance), c(7, 4))
}) })
test_that("xgb.plot.tree works", { test_that("xgb.plot.tree works", {