From 5773d4d3c40822d22b007032d223305c19cc577b Mon Sep 17 00:00:00 2001 From: hetong007 Date: Thu, 27 Aug 2015 16:02:41 -0700 Subject: [PATCH] fix test --- R-package/tests/testthat/test_basic.R | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/R-package/tests/testthat/test_basic.R b/R-package/tests/testthat/test_basic.R index 1c4a85921..791f1246c 100644 --- a/R-package/tests/testthat/test_basic.R +++ b/R-package/tests/testthat/test_basic.R @@ -2,19 +2,18 @@ require(xgboost) context("basic functions") -test_that("data loading", { - data(agaricus.train, package='xgboost') - data(agaricus.test, package='xgboost') -}) +data(agaricus.train, package='xgboost') +data(agaricus.test, package='xgboost') +train = agaricus.train +test = agaricus.test -test_that("train and prediction",{ - train = agaricus.train - test = agaricus.test +test_that("train and predict", { bst = xgboost(data = train$data, label = train$label, max.depth = 2, eta = 1, nthread = 2, nround = 2, objective = "binary:logistic") pred = predict(bst, test$data) }) + test_that("early stopping", { res = xgb.cv(data = train$data, label = train$label, max.depth = 2, nfold = 5, eta = 0.3, nthread = 2, nround = 20, objective = "binary:logistic",