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

Increase cover of tests #Rstat

Browse Source
This commit is contained in:
pommedeterresautee 2015-12-02 10:40:15 +01:00
parent d04f7005de
commit 1678a6fbdb
1 changed files with 22 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,50 +14,55 @@ df[,AgeCat := as.factor(ifelse(Age > 30, "Old", "Young"))]
df[,ID := NULL] df[,ID := NULL]
sparse_matrix <- sparse.model.matrix(Improved~.-1, data = df) sparse_matrix <- sparse.model.matrix(Improved~.-1, data = df)
output_vector <- df[,Y := 0][Improved == "Marked",Y := 1][,Y] output_vector <- df[,Y := 0][Improved == "Marked",Y := 1][,Y]
bst <- xgboost(data = sparse_matrix, label = output_vector, max.depth = 9, bst.Tree <- xgboost(data = sparse_matrix, label = output_vector, max.depth = 9,
eta = 1, nthread = 2, nround = 10, objective = "binary:logistic") eta = 1, nthread = 2, nround = 10, objective = "binary:logistic", booster = "gbtree")
bst.GLM <- xgboost(data = sparse_matrix, label = output_vector,
eta = 1, nthread = 2, nround = 10, objective = "binary:logistic", booster = "gblinear")
feature.names <- agaricus.train$data@Dimnames[[2]] feature.names <- agaricus.train$data@Dimnames[[2]]
test_that("xgb.dump works", { test_that("xgb.dump works", {
capture.output(print(xgb.dump(bst))) capture.output(print(xgb.dump(bst.Tree)))
expect_true(xgb.dump(bst, 'xgb.model.dump', with.stats = T)) capture.output(print(xgb.dump(bst.GLM)))
expect_true(xgb.dump(bst.Tree, 'xgb.model.dump', with.stats = T))
}) })
test_that("xgb.model.dt.tree works with and without feature names", { test_that("xgb.model.dt.tree works with and without feature names", {
names.dt.trees <- c("ID", "Feature", "Split", "Yes", "No", "Missing", "Quality", "Cover", 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") "Tree", "Yes.Feature", "Yes.Cover", "Yes.Quality", "No.Feature", "No.Cover", "No.Quality")
dt.tree <- xgb.model.dt.tree(feature_names = feature.names, model = bst) dt.tree <- xgb.model.dt.tree(feature_names = feature.names, model = bst.Tree)
expect_equal(names.dt.trees, names(dt.tree)) expect_equal(names.dt.trees, names(dt.tree))
expect_equal(dim(dt.tree), c(162, 15)) expect_equal(dim(dt.tree), c(162, 15))
xgb.model.dt.tree(model = bst) xgb.model.dt.tree(model = bst.Tree)
}) })
test_that("xgb.importance works with and without feature names", { test_that("xgb.importance works with and without feature names", {
importance <- xgb.importance(feature_names = sparse_matrix@Dimnames[[2]], model = bst) importance.Tree <- xgb.importance(feature_names = sparse_matrix@Dimnames[[2]], model = bst.Tree)
expect_equal(dim(importance), c(7, 4)) expect_equal(dim(importance.Tree), c(7, 4))
expect_equal(colnames(importance), c("Feature", "Gain", "Cover", "Frequency")) expect_equal(colnames(importance.Tree), c("Feature", "Gain", "Cover", "Frequency"))
xgb.importance(model = bst) xgb.importance(model = bst.Tree)
xgb.plot.importance(importance_matrix = importance.Tree)
}) })
test_that("xgb.importance works with GLM model", { test_that("xgb.importance works with GLM model", {
bst.GLM <- xgboost(data = sparse_matrix, label = output_vector,
eta = 1, nthread = 2, nround = 10, objective = "binary:logistic", booster = "gblinear")
importance.GLM <- xgb.importance(feature_names = sparse_matrix@Dimnames[[2]], model = bst.GLM) importance.GLM <- xgb.importance(feature_names = sparse_matrix@Dimnames[[2]], model = bst.GLM)
expect_equal(dim(importance.GLM), c(10, 2)) expect_equal(dim(importance.GLM), c(10, 2))
expect_equal(colnames(importance.GLM), c("Feature", "Weight")) expect_equal(colnames(importance.GLM), c("Feature", "Weight"))
xgb.importance(model = bst.GLM) xgb.importance(model = bst.GLM)
xgb.plot.importance(importance.GLM)
}) })
test_that("xgb.plot.tree works with and without feature names", { test_that("xgb.plot.tree works with and without feature names", {
xgb.plot.tree(feature_names = feature.names, model = bst) xgb.plot.tree(feature_names = feature.names, model = bst.Tree)
xgb.plot.tree(model = bst) xgb.plot.tree(model = bst.Tree)
}) })
test_that("xgb.plot.multi.trees works with and without feature names", { test_that("xgb.plot.multi.trees works with and without feature names", {
xgb.plot.multi.trees(model = bst, feature_names = feature.names, features.keep = 3) xgb.plot.multi.trees(model = bst.Tree, feature_names = feature.names, features.keep = 3)
xgb.plot.multi.trees(model = bst, features.keep = 3) xgb.plot.multi.trees(model = bst.Tree, features.keep = 3)
}) })
test_that("xgb.plot.deepness works", { test_that("xgb.plot.deepness works", {
xgb.plot.deepness(model = bst) xgb.plot.deepness(model = bst.Tree)
}) })