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

R accessors for model attributes

Browse Source
This commit is contained in:
Vadim Khotilovich
2016-05-02 00:20:44 -05:00
parent 0839aed380
commit 79c7c9e5bb
4 changed files with 140 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -28,6 +28,17 @@ test_that("xgb.dump works", {
expect_true(xgb.dump(bst.Tree, 'xgb.model.dump', with.stats = T))
})
test_that("xgb.attr", {
val <- "my attribute value"
expect_error(xgb.attr(bst.Tree, NULL))
expect_error(xgb.attr(val, val))
xgb.attr(bst.Tree, "my_attr") <- val
expect_equal(xgb.attr(bst.Tree, "my_attr"), val)
xgb.save(bst.Tree, 'xgb.model')
bst1 <- xgb.load('xgb.model')
expect_equal(xgb.attr(bst1, "my_attr"), val)
})
test_that("xgb.model.dt.tree works with and without feature names", {
names.dt.trees <- c("ID", "Feature", "Split", "Yes", "No", "Missing", "Quality", "Cover",
"Tree", "Yes.Feature", "Yes.Cover", "Yes.Quality", "No.Feature", "No.Cover", "No.Quality")