diff --git a/src/data/iterative_dmatrix.cc b/src/data/iterative_dmatrix.cc index 30583a943..83287dccf 100644 --- a/src/data/iterative_dmatrix.cc +++ b/src/data/iterative_dmatrix.cc @@ -244,7 +244,8 @@ BatchSet IterativeDMatrix::GetGradientIndex(BatchParam const& ghist_ = std::make_shared(&ctx_, Info(), *ellpack_, param); } - if (param.sparse_thresh != tree::TrainParam::DftSparseThreshold()) { + if (!std::isnan(param.sparse_thresh) && + param.sparse_thresh != tree::TrainParam::DftSparseThreshold()) { LOG(WARNING) << "`sparse_threshold` can not be changed when `QuantileDMatrix` is used instead " "of `DMatrix`."; }