sync Mar 27 2023

2023-03-27 18:54:41 +02:00
parent 8c77e936d1 401ce5cf5e
commit c50cc424bc
43 changed files with 643 additions and 508 deletions
--- a/tests/cpp/gbm/test_gbtree.cc
+++ b/tests/cpp/gbm/test_gbtree.cc
@@ -505,7 +505,7 @@ TEST(GBTree, PredictRange) {
    auto h_out_predt_full = out_predt->HostVector();

    ASSERT_TRUE(std::equal(h_out_predt.begin(), h_out_predt.end(), h_out_predt_full.begin()));
-
+    // Out of range.
    ASSERT_THROW(learner->InplacePredict(x, PredictionType::kValue,
                                         std::numeric_limits<float>::quiet_NaN(), &out_predt, 0, 3),
                 dmlc::Error);