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

Update test to take care of API change

Browse Source
This commit is contained in:
pommedeterresautee 2015-11-30 14:08:27 +01:00
parent c5dedeb318
commit ad8766dfa4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23,7 +23,7 @@ test_that("xgb.dump works", {
test_that("xgb.importance works", { test_that("xgb.importance works", {
expect_true(xgb.dump(bst, 'xgb.model.dump', with.stats = T)) expect_true(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]], model = bst)
expect_equal(dim(importance), c(7, 4)) expect_equal(dim(importance), c(7, 4))
expect_equal(colnames(importance), c("Feature", "Gain", "Cover", "Frequency")) expect_equal(colnames(importance), c("Feature", "Gain", "Cover", "Frequency"))
}) })