fixed unit test in R
This commit is contained in:
parent
92b996513e
commit
339a53d9d4
@ -13,7 +13,7 @@ test_that("glm works", {
|
|||||||
nthread = 2, alpha = 0.0001, lambda = 1)
|
nthread = 2, alpha = 0.0001, lambda = 1)
|
||||||
watchlist <- list(eval = dtest, train = dtrain)
|
watchlist <- list(eval = dtest, train = dtrain)
|
||||||
num_round <- 2
|
num_round <- 2
|
||||||
expect_that(bst <- xgb.train(param, dtrain, num_round, watchlist), not(throws_error()))
|
bst <- xgb.train(param, dtrain, num_round, watchlist)
|
||||||
expect_that(ypred <- predict(bst, dtest), not(throws_error()))
|
ypred <- predict(bst, dtest)
|
||||||
expect_equal(length(getinfo(dtest, 'label')), 1611)
|
expect_equal(length(getinfo(dtest, 'label')), 1611)
|
||||||
})
|
})
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user