Fix R dart prediction. (#5204)

* Fix R dart prediction and add test.
2020-01-16 12:11:04 +08:00
parent 808f61081b
commit 5199b86126
5 changed files with 80 additions and 35 deletions
--- a/R-package/tests/testthat/test_helpers.R
+++ b/R-package/tests/testthat/test_helpers.R
@@ -157,7 +157,7 @@ test_that("SHAPs sum to predictions, with or without DART", {
      params = c(
        list(
          booster = booster,
-          objective = "reg:linear",
+          objective = "reg:squarederror",
          eval_metric = "rmse"),
        if (booster == "dart")
          list(rate_drop = .01, one_drop = T)),