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

some fixes for Travis #Rstat

Browse Source
This commit is contained in:
pommedeterresautee
2015-11-30 15:47:10 +01:00
parent c09c02300a
commit 730bd72056
9 changed files with 25 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,13 +23,13 @@ test_that("xgb.dump works", {
})
test_that("xgb.importance works", {
importance <- xgb.importance(sparse_matrix@Dimnames[[2]], model = bst)
importance <- xgb.importance(feature_names = sparse_matrix@Dimnames[[2]], model = bst)
expect_equal(dim(importance), c(7, 4))
expect_equal(colnames(importance), c("Feature", "Gain", "Cover", "Frequency"))
})
test_that("xgb.plot.tree works", {
xgb.plot.tree(names = agaricus.train$data@Dimnames[[2]], model = bst)
xgb.plot.tree(feature_names = agaricus.train$data@Dimnames[[2]], model = bst)
})
test_that("xgb.plot.deepness works", {
@@ -37,5 +37,5 @@ test_that("xgb.plot.deepness works", {
})
test_that("xgb.plot.multi.trees works", {
xgb.plot.multi.trees(model = bst, names = agaricus.train$data@Dimnames[[2]], 3)
xgb.plot.multi.trees(model = bst, feature_names = agaricus.train$data@Dimnames[[2]], 3)
})