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

TST: Added R unit test for glm

Browse Source
This commit is contained in:
terrytangyuan 2015-09-05 22:50:27 -04:00
parent 3d6c831e8a
commit 92b996513e
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
R-package/tests/testthat/test_glm.R Normal file
View File

@ -0,0 +1,19 @@
context('Test generalized linear models')
require(xgboost)
test_that("glm works", {
data(agaricus.train, package='xgboost')
data(agaricus.test, package='xgboost')
dtrain <- xgb.DMatrix(agaricus.train$data, label = agaricus.train$label)
dtest <- xgb.DMatrix(agaricus.test$data, label = agaricus.test$label)
expect_equal(class(dtrain), "xgb.DMatrix")
expect_equal(class(dtest), "xgb.DMatrix")
param <- list(objective = "binary:logistic", booster = "gblinear",
nthread = 2, alpha = 0.0001, lambda = 1)
watchlist <- list(eval = dtest, train = dtrain)
num_round <- 2
expect_that(bst <- xgb.train(param, dtrain, num_round, watchlist), not(throws_error()))
expect_that(ypred <- predict(bst, dtest), not(throws_error()))
expect_equal(length(getinfo(dtest, 'label')), 1611)
})